Método de aumento de dados por geração conjunta de imagens celulares e máscaras baseado em modelo de difusão

XU Kaige ,  

SUN Xiaoming ,  

CHEN Yongji ,  

CHEN Yan ,  

LONG Bing ,  

HAN Yongqi ,  

SU Bochao ,  

WEI Changning ,  

摘要

Para resolver o problema da insuficiência de dados rotulados na tarefa de segmentação de imagens celulares e as limitações dos métodos tradicionais de aumento, este estudo propõe inicialmente um método de aumento de dados baseado em modelo de difusão para geração conjunta de imagens celulares e máscaras. O método constrói uma estrutura de geração conjunta composta por uma U-Net de previsão de ruído e um amostrador de ruído, empilhando as imagens celulares e máscaras por canal para formar uma imagem conjunta de quatro canais, realizando geração conjunta em estágio único de dados sintéticos de alta qualidade, evitando erros introduzidos pela geração em múltiplos estágios. O modelo integra incorporação temporal e mecanismo de autoatenção de múltiplas cabeças para reforçar a capacidade de modelagem da relação entre características estruturais celulares e máscaras correspondentes, e adota função de perda de erro quadrático médio e agendamento de taxa de aprendizado com recozimento cossenoidal para otimizar o processo de treinamento. Resultados experimentais abrangentes nos conjuntos de dados de referência CryoNuSeg e ISBI2012 mostram que o método proposto melhora significativamente o desempenho dos modelos de segmentação a jusante. Sob a configuração de amostragem DPM++ 2M Karras, os valores de IoU e Dice para CryoNuSeg atingiram 62,50% e 75,78%, respectivamente, superando métodos tradicionais de aumento como inversão e rotação. Os resultados verificam a superioridade do aumento por geração conjunta na expansão da diversidade de dados e na melhoria da precisão da segmentação, fornecendo uma solução eficiente e viável para aumento de dados na segmentação de imagens celulares em cenários com escassez de anotações.

关键词

segmentação de imagens celulares; modelo de difusão; aumento de dados; UNet; geração conjunta

阅读全文

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

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