arrow
返回

Diverse Data Generation for Retinal Layer Segmentation With Potential Structure Modeling

delete2024-10-01
delete1
PRE
AI
K
Kun Huang
X
Xiao Ma
张
张泽天 (Zetian Zhang)
Y
Yuhan Zhang
S
Songtao Yuan
H
Huazhu Fu *
Q
Qiang Chen *
DOI:10.1109/TMI.2024.3384484delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Accurate retinal layer segmentation on optical coherence tomography (OCT) images is hampered by the challenges of collecting OCT images with diverse pathological characterization and balanced distribution. Current generative models can produce high-realistic images and corresponding labels without quantitative limitations by fitting distributions of real collected data. Nevertheless, the diversity of their generated data is still limited due to the inherent imbalance of training data. To address these issues, we propose an image-label pair generation framework that generates diverse and balanced potential data from imbalanced real samples. Specifically, the framework first generates diverse layer masks, and then generates plausible OCT images corresponding to these layer masks using two customized diffusion probabilistic models respectively. To learn from imbalanced data and facilitate balanced generation, we introduce pathological-related conditions to guide the generation processes. To enhance the diversity of the generated image-label pairs, we propose a potential structure modeling technique that transfers the knowledge of diverse sub-structures from lowly- or non-pathological samples to highly pathological samples. We conducted extensive experiments on two public datasets for retinal layer segmentation. Firstly, our method generates OCT images with higher image quality and diversity compared to other generative methods. Furthermore, based on the extensive training with the generated OCT images, downstream retinal layer segmentation tasks demonstrate improved results. The code is publicly available at: https://github.com/nicetomeetu21/GenPSM.
Keyword:
Image segmentation
Retina
Data models
Diseases
Pathology
Training
Task analysis
Optical coherence tomography
layer segmentation
diffusion probabilistic model
diverse data generation
contrast learning

期刊

IEEE Transactions on Medical Imaging 封面图
IEEE Transactions on Medical Imaging
IF:
9.8
论文数:
6.2K
被引数:
3.7W

机构

N
Nanjing Medical University
学者数:
4.7W
论文数: 2.8W
被引数: 3.5K
C
Chinese University of Hong Kong
学者数:
3.4W
论文数: 3.2W
被引数: 5.6W
C
cuhk shenzhen research institute
学者数:
768
论文数: 486
被引数: 1
学者 查看更多机构
引用论文

引用论文

Self-Supervised Tumor Segmentation With Sim2Real Adaptation
err2023-09-01
err3
PREAI
errZhang, Xiaoman; Xie, Weidi; Huang, Chaoqin; Zhang, Ya; Chen, Xin; Tian, Qi; Wang, Yanfeng
err分享
err收藏
Psychosocial outcomes of children with ear infections and hearing problems: a longitudinal study
err2014-03-04
err0
errOAAI
errAnthony Hogan; Rebecca L Phillips; Damien Howard; Vasoontara Yiengprugsawan
err分享
err收藏
Retinal and Choroidal Thinning-A Predictor of Coronary Artery Occlusion
err2022-08-20
err22
errOAAI
errMatuleviciute, Indre; Sidaraite, Agne; Tatarunas, Vacis; Veikutiene, Audrone; Dobiliene, Olivija; Zaliuniene, Dalia
err分享
err收藏
Provably safe motion of mobile robots in human environments
err2017-09-01
err0
errOAAI
errStefan B. Liu; Hendrik Roehm; Christian Heinzemann; Ingo Lutkebohle; Jens Oehlerking; Matthias Althoff
err分享
err收藏
Crystal and magnetic structures of hexagonal YMnO3
err2013-11-18
err0
PREAI
errChristopher J. Howard; Branton J. Campbell; Harold T. Stokes; Michael A. Carpenter; Richard I. Thomson
err分享
err收藏
Level of psychosocial adaptation in young school children with otitis media
err2007-12-01
err0
PREAI
errAngelique Timmerman; Cor Meesters; Lucien Anteunis; Mickey Chenault
err分享
err收藏
学者 查看更多内容