LAVIFT: Latent-Action-Guided Vision Fine-Tuning for Surgical Interaction Recognition

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, extended

Summary

LAViFiT is an end-to-end latent-action-guided framework designed for vision-language fine-tuning in surgical interaction recognition. It addresses challenges in adapting pretrained vision-language models and encoders to fine-grained surgical tasks, specifically the issues of noisy representations from frozen encoders and limited spatial precision from full fine-tuning. The framework integrates an inverse dynamics model to capture action-induced visual changes and a forward world model to guide the encoder towards action-relevant regions. A novel Patch-level SIG Regularizer prevents local feature collapse without requiring additional supervision like bounding boxes. Experiments on CholecT50 and ProstaTD datasets demonstrate that LAViFiT improves recognition and image-text alignment across various encoders, including ViT-L/14, DINOv2-L/14, and SurgeNet-L/14. The SurgeNet-L model achieved the best overall results, with 18.50 mAP on CholecT50 and 27.51 mAP on ProstaTD. The method also yields stronger grounding in instrument-tissue interaction regions and more spatially coherent features. The source code is publicly available.

Key takeaway

For AI Scientists and Machine Learning Engineers developing surgical AI systems, if you are struggling with adapting general-domain vision models to fine-grained instrument-tissue interaction recognition, you should consider implementing the LAViFiT framework. This method significantly improves recognition performance and feature grounding by explicitly guiding your vision encoder to focus on action-relevant regions. You can achieve more spatially coherent and interpretable features, but be aware that world model objectives with SIG-style regularization can be sensitive to hyperparameter choices.

Key insights

Latent-action modeling with patch-level regularization precisely fine-tunes vision encoders for surgical instrument-tissue interaction recognition.

Principles

Method

LAViFiT jointly trains a vision encoder, Delta-IDM, DiT-style FWM, clip aggregator, and text encoder. It uses classification, forward prediction, and state/patch-level SIG regularization losses.

In practice

Topics

Best for: Computer Vision Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, Robotics Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.