結直腸癌是全球最常見的癌症之一,早期發現和切除息肉可以有效降低死亡率。自動息肉分割作為計算機輔助診斷的核心技術,能夠幫助醫生快速定位病變區域。然而,構建高性能的分割模型通常需要來自多家醫院的大規模標註數據,而醫療數據的高度敏感性使得集中式訓練面臨嚴重的隱私合規挑戰。聯邦學習提供了在保護數據隱私的前提下進行協作訓練的解決方案,但反覆交換全精度模型參數的通信開銷成為部署瓶頸。針對這一痛點,研究團隊提出了QFedPolyp——一種通信與推理高效的聯邦學習框架。
QFedPolyp的核心思想是結合量化感知訓練(QAT)與低精度通信。在訓練階段,每家醫院在本地使用輕量級U-Net架構,並引入QAT技術,使模型在訓練過程中學習適應量化噪聲。具體而言,在前向傳播中模擬權重量化,而在反向傳播中仍使用全精度梯度,從而保證模型在量化後的性能。訓練完成後,客户端將量化後的8位精度參數發送至中央服務器,服務器使用聯邦平均算法進行聚合,並將更新後的全局模型分發給各客户端。這種方案將通信數據量減少了約4倍,同時保持模型精度幾乎不變。實驗在四個息肉分割數據集上驗證了其有效性:在Kvasir-SEG和CVC-ClinicVideoDB上,全精度聯邦訓練的Dice係數分別達到0.910和0.930,而8位量化版本僅損失不到0.5%的精度。此外,量化後的模型由於權重以低精度存儲,推理時計算複雜度降低,推理速度提升高達1.5倍,這對實時臨牀場景至關重要。
QFedPolyp的提出不僅解決了聯邦學習在醫療圖像分析中的通信效率問題,還通過量化加速了推理過程,使得輕量級模型能夠在邊緣設備上快速部署。這項研究發表於arXiv,作者為Madan Baduwal等人。未來,該框架有望擴展到其他醫學圖像分割任務,並進一步優化量化策略以平衡精度與效率。總體而言,QFedPolyp為隱私保護的協作醫療AI提供了一條切實可行的路徑,具有重要的臨牀應用價值。