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.

Ссылки и действия