基于标签自动校准的变温自蒸馏脑肿瘤MRI图像分类方法
作者:
作者单位:

上海理工大学健康科学与工程学院,上海 200093

作者简介:

通讯作者:

基金项目:

国家科学技术部项目(G2021013008);中国高校产学研创新基金(2023RY011);上海理工大学医工交叉重点创新项目(1022308502);华为AI算力加速计划项目;上海理工大学教师发展研究重点项目(CFTD2025ZD08)。


Classification Method of Variable-Temperature Self-distillation Brain Tumor MRI Images Based on Label-Auto-calibration
Author:
Affiliation:

1School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China

Fund Project:

Project of the Ministry of Science and Technology (No.G2021013008); China University Industry-University-Research Innovation Fund (No.2023RY011); Medical and Engineering Interdisciplinary Key Innovation Project of University of Shanghai for Science and Technology (No.1022308502); Huawei AI Computing Power Acceleration Program; Key Project of Teacher Development Research at University of Shanghai for Science and Technology (No.CFTD2025ZD08).

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    脑肿瘤是神经系统中最常见的疾病之一,磁共振成像(Magnetic resonance imaging,MRI)检查是常见的脑肿瘤筛查方法。深度学习模型在MRI图像识别上虽已具备较高性能,但在计算资源受限场景下运行效果不理想,且其性能在剪枝轻量化后易出现下降,为此本研究聚焦于提高轻量化模型的脑肿瘤MRI图像分类性能,针对上述挑战提出一种基于标签自动校准的变温自蒸馏脑肿瘤MRI图像分类方法。研究从轻量化模型性能提升的核心需求出发,使用基于标签自动校准的损失函数,避免自蒸馏过程中自教师模型传授错误知识,同时引入一种温度重启的变温蒸馏机制,当验证准确率在最近多轮训练中未提升时,根据当前训练情况重启温度,防止轻量化网络学习难度过大。在多个脑肿瘤数据集上的实验分析结果表明,所提出的自蒸馏方法可有效提高轻量化模型的脑肿瘤识别性能,且性能优于其他典型的自蒸馏方法,具有无需改变模型结构、易于实现的优点。

    Abstract:

    Brain tumors are one of the most common diseases in the nervous system, and magnetic resonance imaging (MRI) examination is a common screening method for brain tumors. Although deep learning models have achieved high performance in MRI image recognition, their operational effectiveness is unsatisfactory in scenarios with limited computational resources, and their performance tends to decline after pruning for lightweight. Therefore, this study focuses on improving the brain tumor MRI image classification performance of lightweight models. To address the aforementioned challenges, classification method of variable-temperature self-distillation brain tumor MRI images based on label-auto-calibration is proposed. Starting from the core requirement of enhancing the performance of lightweight models, this study employs a loss function based on label-auto-calibration loss to avoid imparting incorrect knowledge from the self-teacher model during the self-distillation process. Additionally, a temperature-resetting variable-temperature distillation mechanism is introduced. When the validation accuracy does not improve in recent rounds of training, the temperature is reset based on the current training situation to prevent the lightweight network from encountering excessive learning difficulty. Experimental analysis on multiple brain tumor datasets demonstrates that the proposed self-distillation method can effectively enhance the brain tumor recognition performance of lightweight models, and its performance surpasses other typical self-distillation methods. This method requires no changes to the model structure, is easy to implement, and can be used to enhance the performance of lightweight networks.

    参考文献
    相似文献
    引证文献
引用本文

徐楚迪,何宏,陈家毓,陈宇聪,李泽旭.基于标签自动校准的变温自蒸馏脑肿瘤MRI图像分类方法[J].数据采集与处理,2026,(4):1178-1193

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:2025-10-31
  • 最后修改日期:2026-04-07
  • 录用日期:
  • 在线发布日期: 2026-08-13