提升甲状腺超声结节分类模型在不同设备间的泛化能力
Prototype-Enhanced Multi-View Learning for Thyroid Nodule Ultrasound Classification
- 从多视角学习互补特征,用原型修正决策边界
- 跨设备/域测试中准确率显著优于现有方法
- 适合临床部署的鲁棒性影像诊断系统开发者
甲状腺结节超声分类对早期诊断和临床决策至关重要;然而,现有深度学习方法在分布内数据上表现良好,跨不同超声设备或临床环境时泛化能力有限。这主要源于甲状腺超声图像的显著异质性,导致模型捕捉到虚假相关而非可靠诊断线索。为此,我们提出PEMV-thyroid框架,通过多视角学习互补特征,并结合混合原型信息的原型修正机制优化决策边界,以应对数据异质性。该方法在多个甲状腺超声数据集上进行的大量实验表明,其在跨设备和跨域评估中均持续优于当前最优方法,显著提升真实临床场景下的诊断准确率与泛化性能。源代码已公开于https://github.com/chenyangmeii/Prototype-Enhanced-Multi-View-Learning。
原文摘要 · Abstract (English)
Thyroid nodule classification using ultrasound imaging is essential for early diagnosis and clinical decision-making; however, despite promising performance on in-distribution data, existing deep learning methods often exhibit limited robustness and generalisation when deployed across different ultrasound devices or clinical environments. This limitation is mainly attributed to the pronounced heterogeneity of thyroid ultrasound images, which can lead models to capture spurious correlations rather than reliable diagnostic cues. To address this challenge, we propose PEMV-thyroid, a Prototype-Enhanced Multi-View learning framework that accounts for data heterogeneity by learning complementary representations from multiple feature perspectives and refining decision boundaries through a prototype-based correction mechanism with mixed prototype information. By integrating multi-view representations with prototype-level guidance, the proposed approach enables more stable representation learning under heterogeneous imaging conditions. Extensive experiments on multiple thyroid ultrasound datasets demonstrate that PEMV-thyroid consistently outperforms state-of-the-art methods, particularly in cross-device and cross-domain evaluation scenarios, leading to improved diagnostic accuracy and generalisation performance in real-world clinical settings. The source code is available at https://github.com/chenyangmeii/Prototype-Enhanced-Multi-View-Learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。