用分层半监督与自监督融合方法,提升膝骨关节炎分级准确率。
H-SemiS: Hierarchical Fusion of Semi and Self-Supervised Learning for Knee Osteoarthritis Severity Grading
- 将分级任务拆解为二分类子任务,缓解类别不平衡问题。
- 在有限标注数据下,模型在两个数据集上均超越现有方法。
- 适合医疗影像少样本场景下的疾病分级研究者使用。
膝骨关节炎(KOA)是一种退行性关节病,可导致慢性疼痛、活动能力下降和长期残疾。从膝关节X光片自动评估严重程度有助于早期诊断,但现有方法严重依赖大量标注数据,且对类别不平衡、噪声样本和临床标注差异敏感。为此,我们提出一种分层半监督与自监督融合框架(H-SemiS),用于在标注数据有限的情况下进行膝骨关节炎严重程度分级。H-SemiS不将分级视为单一多分类任务,而是在半监督师生架构中将其分解为一系列二分类子任务,有效缓解类别不平衡问题。为进一步增强无标签数据的特征学习,框架引入对抗性自监督重建模块,促使网络捕捉稳健的解剖结构。同时,采用受量子启发的特征混合师生设计,在伪标签存在噪声时优化相邻等级间的判别边界。我们在两个具有挑战性的多分类数据集上全面评估H-SemiS,并在两个二分类数据集上测试其泛化能力。实验结果表明,该框架在多个评价指标上持续优于多种竞争基线及先进方法。代码已公开于 https://github.com/chandravardhan-singh-raghaw/H-SemiS。
原文摘要 · Abstract (English)
Knee osteoarthritis (KOA) is a degenerative joint disease that can lead to chronic pain, reduced mobility, and long-term disability. Automated severity grading from knee radiographs can support early assessment, but current methods heavily depend on large labeled datasets and remain sensitive to class imbalance, noisy samples, and variability in clinical annotations. To alleviate these limitations, we propose a Hierarchical fusion of Semi-Supervised framework with Self-Supervision (H-SemiS) for KOA severity grading in knee X-ray samples using limited annotated data. Rather than treating severity grading as a flat multi-class problem, H-SemiS decomposes the task into a sequence of binary sub-tasks within a semi-supervised teacher-student architecture, directly mitigating the impact of class imbalance. To further enhance feature learning from unlabeled data, the framework integrates an adversarial self-supervised reconstruction module that encourages the network to capture robust anatomical structures. In parallel, a teacher-student design with quantum-inspired feature mixing improves discrimination boundaries between adjacent grades when pseudo-labels are noisy. We comprehensively evaluate H-SemiS on two challenging multi-class datasets and assess its generalizability on two binary-class datasets. Our experimental results demonstrate the superiority of the proposed H-SemiS framework across multiple evaluation metrics, consistently outperforming several competing baselines and state-of-the-art methods. The code is publicly available at https://github.com/chandravardhan-singh-raghaw/H-SemiS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。