Home Research AI for Healthcare TrajPred

TrajPred

Trajectory-Conditioned Joint Embedding Prediction for Surgical Instrument-Tissue Interaction Recognition in Vision-Language Models — IROS 2026

TrajPred encodes instrument trajectories to inject temporal motion cues into Vision-Language Models, then uses a trajectory-conditioned predictor to generate visual semantic embeddings that capture fine-grained action details. With prompt tuning and a verb-rephrasing technique for task adaptation, TrajPred improves instrument–tissue interaction recognition on CholecT50 in both Average Precision and Top-K accuracy.

📄 Paper (arXiv) 💻 Code

Abstract

Recognizing instruments' interactions with tissues is essential for building context-aware AI assistants in robotic surgery. Vision-language models (VLMs) have opened a new avenue for surgical perception and achieved better generalization on a wide range of tasks compared to conventional task-specific deep learning approaches. However, their performance on instrument–tissue interaction recognition remains limited, largely due to two challenges: (1) many models do not effectively leverage temporal information, and (2) alignment between vision and text often misses fine-grained action details. To address these issues, we propose TrajPred, a framework that encodes instrument trajectories to incorporate temporal motion cues; and conditioned on these trajectories, introduces a predictor module to generate visual semantic embeddings that better capture fine-grained action details. We further incorporate prompt tuning and a verb-rephrasing technique to enable smooth adaptation to the instrument–tissue interaction recognition task. Extensive experiments on the public laparoscopic benchmark, CholecT50, show that our method improves both Average Precision and Top-K accuracy.

We also investigate whether visual embeddings of instrument–tissue interaction regions align better with the corresponding text by visualizing the cosine similarity between visual and textual embeddings. The visualization results indicate that the proposed method improves alignment between relevant visual and textual representations.


Figures

TrajPred experimental results
Image–text similarity heatmaps over six consecutive frames, where red indicates higher cosine similarity. Similarity is computed between predictor-generated spatiotemporal tubelet tokens and the embedding of the ground-truth triplet (when it appears in the top-5 predictions). Each heatmap represents a two-frame temporal unit and is overlaid on the second frame. Top-5 triplet scores are shown alongside the heatmaps. Green and orange bounding boxes denote the grasper and bipolar instruments, respectively.

Citation

@article{cheng2026trajpred,
  title={TrajPred: Trajectory-Conditioned Joint Embedding Prediction for Surgical Instrument-Tissue Interaction Recognition in Vision-Language Models},
  author={Cheng, Jiajun and Yu, Xiaofan and Tripathi, Subarna and Liu, Sainan and Lin, Shan},
  journal={arXiv preprint arXiv:2603.06999},
  year={2026}
}

Authors

Xiaofan Yu
University of California, Merced
Sainan Liu
Intel
Shan Lin
Shan Lin
MARGIN Lab, ASU

Update — Metrics Bug Fix

We identified a bug in our metrics implementation that only affects the CLIP-VIT-L-Patch14-FT (vision encoder frozen) row of Table 2 in the paper. The fix does not change the paper's main claims.

Overall Average Precision (AP) and Top-K performance with K = |GT| under the RDV split and the unseen-verb setting on CholecT50. ‡ denotes surgical-domain pretrained VLMs. All VL-JEPA variants and our methods use a frozen visual encoder.
Setting Method API APV APT APIVT Top@|GT| Top@5 Top@10 Top@20
Standard RDV CLIP-VIT-L-Patch14-FT (vision encoder frozen) 61.1744.4329.6711.0255.3277.1288.0493.60
CLIP-VIT-L-Patch14-Full-FT 84.1254.4334.6713.1261.2479.5087.8494.42
CLIP-VIT-L-Patch14-GOAL-FT 76.0445.9332.0813.1561.0780.9387.9995.04
SurgVLP-Full-FT 71.1847.7033.3512.6859.0281.9888.7294.93
HecVL-Full-FT 76.0750.9232.0912.6758.0378.8086.5993.58
PeskaVL-Full-FT 75.2048.7130.2912.1257.0576.7686.4393.54
VL-JEPA (Image) 72.5149.2231.5812.6560.0681.5890.0495.31
VL-JEPA (Video) 82.9555.1734.1613.4961.9179.8688.8194.91
TrajPred (Ours) 86.3759.2336.0114.7765.4584.3391.6197.02
Unseen Verb SurgVLP-Full-FT 54.5735.3221.968.0915.1722.9829.0733.12
HecVL-Full-FT 60.5235.2723.768.6816.6925.5530.6134.15
PeskaVL-Full-FT 57.9235.7923.708.5417.7326.9731.9534.91
VL-JEPA (Image) 53.7332.0423.948.2817.5325.6532.2934.85
VL-JEPA (Video) 62.3335.9724.889.0219.3927.6732.9335.21
TrajPred (Ours) 74.3143.6226.3311.2620.8828.1432.9936.27