用部分标注数据训练多疾病眼底筛查模型,提升泛化能力。
PSScreen: Partially Supervised Multiple Retinal Disease Screening
- 双流结构分别学习确定性与概率特征,通过不确定性注入增强表达。
- 利用文本引导解耦特征并蒸馏对齐,提升跨域泛化性能。
- 伪标签一致性与自蒸馏机制缓解缺失标签问题,适合医疗多病种筛查。
利用多个部分标注的数据集训练多疾病眼底筛查模型,可减少对全标注数据的依赖,但因不同医疗站点间存在显著域偏移,且部分类别无标签,仍具挑战。为此,我们提出PSScreen,一种新型的部分监督多疾病眼底筛查模型。PSScreen采用双流架构,一侧重学习确定性特征,另一侧重通过不确定性注入学习概率特征。随后,利用文本引导将两类特征解耦为疾病特异性特征,并通过特征蒸馏对齐以增强域泛化能力。同时,通过双流间伪标签一致性解决标签缺失问题,并引入自蒸馏机制,将确定性流中已知类别的任务相关语义传递至概率流,进一步提升检测性能。实验表明,PSScreen在六种眼底疾病及正常状态上平均显著提升检测性能,在域内和域外数据集上均达到先进水平。代码已公开于https://github.com/boyiZheng99/PSScreen。
原文摘要 · Abstract (English)
Leveraging multiple partially labeled datasets to train a model for multiple retinal disease screening reduces the reliance on fully annotated datasets, but remains challenging due to significant domain shifts across training datasets from various medical sites, and the label absent issue for partial classes. To solve these challenges, we propose PSScreen, a novel Partially Supervised multiple retinal disease Screening model. Our PSScreen consists of two streams and one learns deterministic features and the other learns probabilistic features via uncertainty injection. Then, we leverage the textual guidance to decouple two types of features into disease-wise features and align them via feature distillation to boost the domain generalization ability. Meanwhile, we employ pseudo label consistency between two streams to address the label absent issue and introduce a self-distillation to transfer task-relevant semantics about known classes from the deterministic to the probabilistic stream to further enhance the detection performances. Experiments show that our PSScreen significantly enhances the detection performances on six retinal diseases and the normal state averagely and achieves state-of-the-art results on both in-domain and out-of-domain datasets. Codes are available at https://github.com/boyiZheng99/PSScreen.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。