跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

長尾醫學圖像分類的合成數據生成:皮膚病變案例研究

文章摘要

論文提出了一種基於擴散模型的數據增強管道,用於解決醫學圖像分類中的長尾分佈問題。該方法結合修復擴散模型和分佈外後選擇機制,生成多樣化、真實且臨牀有意義的合成樣本。在ISIC2019皮膚病變數據集上,尾部類別性能提升超過28%。

來源arXiv Computer Vision作者: Jiaxiang Jiang, Mahesh Subedar, Omesh Tickoo
長尾醫學圖像分類的合成數據生成:皮膚病變案例研究
報告錯誤

更正渠道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

長尾分佈是醫學圖像分類中一個長期存在的難題。在多類醫學數據集中,某些疾病類別(尤其是罕見但嚴重的疾病)的樣本數量往往遠少於常見類別,導致深度學習模型在“尾部”類別上性能顯著下降。這一問題在臨牀應用中尤為關鍵,因為這些罕見類別往往對應高風險疾病,如某些類型的皮膚癌,需要極高的診斷準確率。

現有的應對方法包括設計專用的網絡架構、採用重平衡的損失函數(如焦點損失)以及手工設計的數據增強策略。然而,這些方法只能帶來有限的性能提升,且由於它們生成的變異有限且大多具有確定性,難以擴展到更復雜的實際場景。

針對上述挑戰,來自英特爾實驗室的Jiaxiang Jiang、Mahesh Subedar和Omesh Tickoo在2026年5月提交的一篇論文中,提出了一種基於擴散模型的合成數據增強管道,專門用於醫學長尾分類任務。該方法的核心是一個新穎的圖像修復擴散模型(inpainting diffusion model),它能夠根據給定的病變區域生成與真實圖像高度相似的合成圖像。此外,該管道還引入了一個分佈外(Out-of-Distribution, OOD)後選擇機制,用於篩選生成的樣本,確保其多樣性和臨牀相關性。

研究人員在ISIC2019皮膚病變分類數據集上進行了實驗驗證。ISIC2019是最大的、最不平衡的醫學影像基準之一,包含大量的皮膚鏡圖像。實驗結果表明,所提出的方法在整體分類性能上取得了顯著提升,尤其是在樣本最少的尾部類別上,準確率提高了超過28%。這一成果充分證明了基於擴散模型的合成數據增強在緩解長尾不平衡問題上的巨大潛力,為提高醫學圖像分類模型的魯棒性提供了新的思路。

該方法的成功不僅適用於皮膚病變分類,還有望推廣到其他醫學影像領域,如視網膜病變、肺部疾病等,為應對醫療數據中普遍存在的長尾分佈問題提供了有效的解決方案。論文作者來自英特爾實驗室,展示了工業界在推動醫療AI發展方面的積極努力。

展開要點與分析

文章情報

工程師進階

要點

  • 長尾分佈是醫學多類數據集中的普遍問題,導致深度學習模型對稀有類別性能不佳。
  • 現有解決方案(特殊架構、重平衡損失、手工數據增強)改進有限且難以擴展。
  • 新方法採用修復擴散模型與分佈外後選擇機制生成合成數據。
  • 在ISIC2019數據集上,樣本最少的類別性能提升超過28%。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。