EndoMatcher: Generalizable Endoscopic Image Matcher via Multi-Domain Pre-training for Robot-Assisted Surgery
2508.05205v1
cs.CV
2025-08-09
Авторы:
Bingyu Yang, Qingyao Tian, Yimeng Geng, Huai Liao, Xinyan Huang, Jiebo Luo, Hongbin Liu
Резюме на русском
Одной из ключевых проблем в области робото-помогаемой хирургии является поиск соответствий в высокоубитых оперативных изображениях. Это необходимо для решения задач типа трехмерного реконструктирования, навигации и понимания сцены операционного поля. Однако эта задача сложна из-за трудных условий визуализации (слабые текстуры, большие перемены угла зрения) и недостатка аннотированных данных. В статье предлагается EndoMatcher — метод, основанный на Vision Transformer, который обучается на большой многодоменной коллекции Endo-Mix6 (1.2M пар изображений с параллельными трансформациями). Для улучшения стабильности обучения и регулирования размеров доменов вводится прогрессивная стратегия обучения. EndoMatcher показал себя эффективно, увеличивая количество совпадающих точек в 2.4 раза по сравнению с текущими лучшими методами. Это демонстрирует возможность обеспечить точные и обобщаемые соответствия в сложных условиях оперативного зрения.
Abstract
Generalizable dense feature matching in endoscopic images is crucial for
robot-assisted tasks, including 3D reconstruction, navigation, and surgical
scene understanding. Yet, it remains a challenge due to difficult visual
conditions (e.g., weak textures, large viewpoint variations) and a scarcity of
annotated data. To address these challenges, we propose EndoMatcher, a
generalizable endoscopic image matcher via large-scale, multi-domain data
pre-training. To address difficult visual conditions, EndoMatcher employs a
two-branch Vision Transformer to extract multi-scale features, enhanced by dual
interaction blocks for robust correspondence learning. To overcome data
scarcity and improve domain diversity, we construct Endo-Mix6, the first
multi-domain dataset for endoscopic matching. Endo-Mix6 consists of
approximately 1.2M real and synthetic image pairs across six domains, with
correspondence labels generated using Structure-from-Motion and simulated
transformations. The diversity and scale of Endo-Mix6 introduce new challenges
in training stability due to significant variations in dataset sizes,
distribution shifts, and error imbalance. To address them, a progressive
multi-objective training strategy is employed to promote balanced learning and
improve representation quality across domains. This enables EndoMatcher to
generalize across unseen organs and imaging conditions in a zero-shot fashion.
Extensive zero-shot matching experiments demonstrate that EndoMatcher increases
the number of inlier matches by 140.69% and 201.43% on the Hamlyn and Bladder
datasets over state-of-the-art methods, respectively, and improves the Matching
Direction Prediction Accuracy (MDPA) by 9.40% on the Gastro-Matching dataset,
achieving dense and accurate matching under challenging endoscopic conditions.
The code is publicly available at https://github.com/Beryl2000/EndoMatcher.
Ссылки и действия
Дополнительные ресурсы: