表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

細い血管のつながりを保ちながらSAMを適応させるTopoLoRA-SAM

TopoLoRA-SAM

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

TopoLoRA-SAMの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1 : TopoLoRA-SAM architecture overview. We freeze the SAM ViT-B image encoder and inject LoRA modules (red) into the feed-forward network (FFN) layers of each transformer block. A lightweight depthwise-separable convolutional adapter (green) refines the high-resolution embedding tensor before mask decoding. Training uses a topology-aware loss combining BCE, Dice, and clDice.

出典:TopoLoRA-SAM / arXiv ↗

概要

  • SAMの画像エンコーダーを固定し、LoRA、局所調整部品、中心線を考慮する学習目標で二値分割へ適応する。
  • 著者らの五データセット比較では、眼底平均Diceが0.595、全体平均が0.735で、学習対象は全パラメータの5.2%だった。
  • 中心線を重視する効果は設定やデータで変わり、固定したSAM本体の推論負荷も残る。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、TopoLoRA-SAMの仕組み

左から右へ読む
  1. 01眼底などの画像
  2. 02SAM画像特徴抽出
  3. 03局所特徴の調整
  4. 04二値領域マスク

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

画像分割は、画像の各画素が血管などの対象に属するかを判定する処理だ。眼底写真の血管は細く、数画素の誤りで枝が途切れる。領域の面積がよく一致していても、血管のつながりは失われ得る。

自然画像で事前学習したSAMを専門画像へ適応させるには、画像の性質の違いにも対応する必要がある。全パラメータの再学習は計算負担が大きいため、更新する部分を絞りながら形を保てるかが実務上の論点になる。

手法

TopoLoRA-SAMは、SAMの画像エンコーダー本体を固定し、各層の一部にLoRAを加える。LoRAは大きな重み行列の変化を小さな行列の組で表す方法で、元の重みを広く更新せずに特徴を調整できる。続いて軽量な畳み込み部品が近隣画素の情報を整え、マスク生成部が対象である確率を画素ごとに出す。

学習時には正解マスクと予測を比べ、画素単位の誤り、領域の重なり、必要に応じて中心線の重なりを損失として使う。中心線は血管を細い線で表したもので、clDiceはその線が正解とどれだけ対応するかを見る指標だ。例えば眼底写真を入れると、血管に当たる画素を示すマスクが得られる。

新規性

事前学習済みの画像分割モデルSAMに対し、少数の追加パラメータで画像の特徴表現を調整し、細い構造の連続性も学習目標に入れた点が提案の中心だ。画像を二種類に分ける二値分割を、眼底写真、内視鏡画像、レーダー画像で検証している。

低ランク適応だけでなく、局所的な形を調整する畳み込み部品と、血管の中心線を評価する損失を組み合わせた。ただし、各部品の効果は同じ大きさではなく、著者らの分析では低ランク適応の寄与が最も大きい。

従来手法との違い

比較対象はU-Net、DeepLabV3+、SegFormer、Mask2Formerの四方式で、同じ五つのデータセットを使った。TopoLoRA-SAMは全体平均と眼底画像の平均Diceで首位だが、各データセットで常に首位という結果ではない。例えばSTAREではMask2FormerのDiceが0.644で、提案法の0.565を上回る。

学習するパラメータは提案法が約490万個、全体の5.2%。Mask2Formerの学習対象約4740万個より少ない一方、SAMの固定部分も推論には必要である。したがって、学習対象の少なさをそのまま実行時の軽さと読み替えることはできない。

実験結果

著者らは眼底血管のDRIVE、STARE、CHASE_DB1、ポリープのKvasir-SEG、海陸判別のSL-SSDDで比較し、三つの乱数設定の平均を報告した。Diceは予測領域と正解領域の重なりを表し、提案法は眼底平均0.595、五データセット平均0.735。比較対象で次点のMask2Formerはそれぞれ0.539、0.709だった。

CHASE_DB1のDiceは提案法0.569、Mask2Former0.285で、提案法の中心線Diceも0.599と報告された。一方、ポリープではMask2Formerが0.933、提案法が0.930だった。これらは論文内の比較条件での著者らの結果であり、別施設の画像で同じ順位になることは示していない。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:眼底写真から血管候補の地図を作り、専門家の確認を助ける用途を想定する。入力は眼底写真、出力は各画素の血管らしさと二値の血管領域である。写真ごとに推定結果を原画像へ重ね、途切れた枝や誤検出を確認できる画面に渡す。

細い血管の連続性が重要なので、面積の重なりだけで結果を選ばず、中心線が保たれているかも確認する。診断や治療判断を自動化する案ではなく、分割結果を人が点検する作業への応用案である。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:利用予定の眼底写真を正解マスク付きで少数集め、撮影条件や症例ごとに分けて学習用と評価用を固定する。TopoLoRA-SAMと、既存の分割方法または論文の比較対象の一つを同じ評価画像で比べる。評価画像は調整に使わない。

Diceに加えて中心線Dice、途切れた血管の件数、誤検出の件数を記録する。元画像へ結果を重ね、担当者が誤りの位置を確認する。画像一枚あたりの処理時間とメモリー使用量も測り、必要な作業時間に収まるか判断する。

限界と課題

著者らは、中心線を重視する損失の効果がデータセットと重みに左右され、重みが大きすぎると性能を損ねると述べる。また、学習する部分を減らしても固定したSAM本体のメモリー使用量と推論負荷は残り、臨床・運用システムとして検証済みではない。

編集上の確認課題は、対象施設の写真で細い枝の欠落や偽の枝がどの程度出るか、解像度や撮影条件の違いに耐えるかである。本文には表と説明で一部の数値に食い違いもあるため、採用判断では元データと再現結果を照合したい。

研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

TopoLoRA-SAM: Topology-Aware Parameter-Efficient Adaptation of Foundation Segmenters for Thin-Structure and Cross-Domain Binary Semantic Segmentation

Foundation segmentation models such as the Segment Anything Model (SAM) exhibit strong zero-shot generalization through large-scale pretraining, but adapting them to domain-specific semantic segmentation remains challenging, particularly for thin structures (e.g., retinal vessels) and noisy modalities (e.g., SAR imagery). Full fine-tuning is computationally expensive and risks catastrophic forgetting. We propose \textbf{TopoLoRA-SAM}, a topology-aware and parameter-efficient adaptation framework for binary semantic segmentation. TopoLoRA-SAM injects Low-Rank Adaptation (LoRA) into the frozen ViT encoder, augmented with a lightweight spatial convolutional adapter and optional topology-aware supervision via differentiable clDice. We evaluate our approach on five benchmarks spanning retinal vessel segmentation (DRIVE, STARE, CHASE\_DB1), polyp segmentation (Kvasir-SEG), and SAR sea/land segmentation (SL-SSDD), comparing against U-Net, DeepLabV3+, SegFormer, and Mask2Former. TopoLoRA-SAM achieves the best retina-average Dice and the best overall average Dice across datasets, while training only \textbf{5.2\%} of model parameters ($\sim$4.9M). On the challenging CHASE\_DB1 dataset, our method substantially improves segmentation accuracy and robustness, demonstrating that topology-aware parameter-efficient adaptation can match or exceed fully fine-tuned specialist models. Code is available at : https://github.com/salimkhazem/Seglab.git

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-30