核心概念
這篇文章提出了一種基於進階神經風格遷移(NST)和可解釋人工智慧(XAI)的新型乳房超音波(BUS)影像增強方法,以解決深度學習模型在有限數據集上過度擬合的問題,並提高模型的可解釋性和效率。
摘要
一種使用進階神經風格遷移的乳房超音波影像增強新方法:一種高效且可解釋的方法
本研究旨在開發一種新穎、高效的乳房超音波 (BUS) 影像增強方法,採用進階神經風格遷移 (NST) 和可解釋人工智慧 (XAI),並利用基於 GPU 的平行基礎架構。
**新型風格損失函數:**結合去神秘化 NST (DNST) 和 mr2NST 模型的風格損失,並使用 ResNet50 模型提取語義特徵,解決過度擬合和梯度消失問題。
**可解釋性:**將基於 XAI 的分層相關性傳播 (LRP) 方法應用於 NST 模型的內容損失函數,以解釋特徵的重要性,並生成熱圖以顯示模型決策過程中的關鍵特徵。
**分散式訓練:**利用 Horovod 框架在 DGX 叢集上的 8 個 GPU 上進行分散式深度學習模型訓練,實現了 5.09 倍的加速,同時保持了模型的準確性。
**影像去噪:**使用 SRAD 濾波器減少增強 BUS 影像中的斑點雜訊,提高影像清晰度和品質。
**影像分類:**使用微調的 ResNet50 模型對去噪後的增強影像進行分類,以提高分類準確性。