arXiv:2512.03054cs.LGcs.AI2025-12

通过自适应冻结编码器,让医疗影像转换更节能且公平。

Energy-Efficient Federated Learning via Adaptive Encoder Freezing for MRI-to-CT Conversion: A Green AI-Guided Research

  • 根据编码器权重变化动态冻结层,减少计算开销
  • 训练能耗和碳排放降低最高达23%,性能几乎不变
  • 适合资源有限的医院参与联合建模,推动健康公平

联邦学习(FL)有望促进医疗平等,使不同机构在数据有限的情况下协同训练深度学习模型。然而,其高资源需求常使计算能力弱的机构无法参与,加剧医疗不公。为此,我们提出一种面向绿色AI的自适应层冻结策略,通过监测每轮编码器权重的相对变化,仅在更新趋稳时冻结层,结合耐心机制避免过早冻结。使用CodeCarbon库追踪能源消耗与碳排放。相比未冻结的对照组,本方法将训练时间、总能耗和二氧化碳当量(CO2eq)排放最多降低23%。同时,磁共振成像(MRI)转计算机断层扫描(CT)的性能保持良好,平均绝对误差(MAE)仅有微小波动。在五种架构中,三类无显著差异,两类甚至有统计学上显著提升。本研究支持临床可接受、环境可持续、社会公平的AI医疗框架,为新型联邦学习评估体系奠定基础。

原文摘要 · Abstract (English)

Federated Learning (FL) holds the potential to advance equality in health by enabling diverse institutions to collaboratively train deep learning (DL) models, even with limited data. However, the significant resource requirements of FL often exclude centres with limited computational infrastructure, further widening existing healthcare disparities. To address this issue, we propose a Green AI-oriented adaptive layer-freezing strategy designed to reduce energy consumption and computational load while maintaining model performance. We tested our approach using different federated architectures for Magnetic Resonance Imaging (MRI)-to-Computed Tomography (CT) conversion. The proposed adaptive strategy optimises the federated training by selectively freezing the encoder weights based on the monitored relative difference of the encoder weights from round to round. A patience-based mechanism ensures that freezing only occurs when updates remain consistently minimal. The energy consumption and CO2eq emissions of the federation were tracked using the CodeCarbon library. Compared to equivalent non-frozen counterparts, our approach reduced training time, total energy consumption and CO2eq emissions by up to 23%. At the same time, the MRI-to-CT conversion performance was maintained, with only small variations in the Mean Absolute Error (MAE). Notably, for three out of the five evaluated architectures, no statistically significant differences were observed, while two architectures exhibited statistically significant improvements. Our work aligns with a research paradigm that promotes DL-based frameworks meeting clinical requirements while ensuring climatic, social, and economic sustainability. It lays the groundwork for novel FL evaluation frameworks, advancing privacy, equity and, more broadly, justice in AI-driven healthcare.

联邦学习绿色AI医疗影像节能

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。