Datenaugmentierungsmethode zur gemeinsamen Generierung von Zellbildern und Masken basierend auf Diffusionsmodell

XU Kaige ,  

SUN Xiaoming ,  

CHEN Yongji ,  

CHEN Yan ,  

LONG Bing ,  

HAN Yongqi ,  

SU Bochao ,  

WEI Changning ,  

摘要

Um das Problem des Mangels an annotierten Daten bei der Zellbildsegmentierungsaufgabe und die Einschränkungen herkömmlicher Augmentierungsmethoden zu lösen, schlägt diese Studie zunächst eine Datenaugmentierungsmethode vor, die auf einem Diffusionsmodell basiert und die gemeinsame Generierung von Zellbildern und Masken umfasst. Die Methode konstruiert einen gemeinsamen Generierungsrahmen, der aus einem Rauschvorhersage-U-Net und einem Rauschsampler besteht, wobei Zellbilder und Masken kanalweise zu einem vierkanaligen gemeinsamen Bild gestapelt werden, um eine einstufige gemeinsame Generierung qualitativ hochwertiger synthetischer Daten zu realisieren und Fehler durch mehrstufige Generierung zu vermeiden. Das Modell integriert Zeit-Embedding und Multi-Head-Selbstaufmerksamkeitsmechanismen, um die Modellierung der Beziehung zwischen Zellstruktureigenschaften und den entsprechenden Masken zu verbessern, und verwendet eine mittlere quadratische Fehlerverlustfunktion sowie einen Cosine-Annealing-Lernratenplan zur Optimierung des Trainingsprozesses. Umfassende experimentelle Ergebnisse auf den Benchmark-Datensätzen CryoNuSeg und ISBI2012 zeigen, dass die vorgeschlagene Methode die Leistung nachgelagerter Segmentierungsmodelle signifikant verbessert. Unter der Sampling-Einstellung DPM++ 2M Karras erreichten CryoNuSeg IoU und Dice 62,50 % bzw. 75,78 % und übertrafen traditionelle Augmentierungsmethoden wie Spiegelung und Rotation. Die Ergebnisse bestätigen die Überlegenheit der gemeinsamen Generierungs-Augmentierung bei der Erweiterung der Datendiversität und der Verbesserung der Segmentierungsgenauigkeit und bieten eine effiziente und praktikable Datenaugmentierungslösung für Zellbildsegmentierung in annotierungsarmen Szenarien.

关键词

Zellbildsegmentierung; Diffusionsmodell; Datenaugmentierung; UNet; gemeinsame Generierung

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website