本文提出了一種自動化MedSAM的方法,通過引入一個輕量級的提示學習模組來替代原有的提示編碼器。該模組可以直接從圖像嵌入中生成相關的提示嵌入,從而實現對特定區域的自動分割。與傳統方法不同,該方法只需要少量的弱標註樣本(緊密邊界框),而不需要完整的分割掩碼。
具體來說,提示學習模組包含兩個分支:一個卷積層生成密集嵌入,一個全連接層生成稀疏嵌入。這兩個嵌入與MedSAM原有的提示嵌入具有相同的形狀,可以直接替換原有的提示編碼器。在訓練過程中,作者設計了一系列損失函數,利用邊界框的緊密性和前景大小等特徵,引導模型學習生成合適的提示嵌入。
作者在三個醫療影像數據集上驗證了該方法的有效性,包括頭部周長、心臟超聲和心臟MRI。結果表明,與需要完整分割掩碼的專門模型相比,該方法在少樣本情況下仍能保持較高的分割精度,且計算效率更高。此外,與現有的自動化SAM方法相比,本文提出的方法也取得了更好的性能。
總之,本文提出的自動化MedSAM方法大幅降低了開發專門分割模型的成本,為醫療影像分析帶來了新的可能性。
Para Outro Idioma
do conteúdo original
arxiv.org
Principais Insights Extraídos De
by Méla... às arxiv.org 10-01-2024
https://arxiv.org/pdf/2409.20293.pdfPerguntas Mais Profundas