Saved in:
Bibliographic Details
Main Authors: Wahd, Assefa, Jaremko, Jacob, Hareendranathan, Abhilash
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2506.17837
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912444006268928
author Wahd, Assefa
Jaremko, Jacob
Hareendranathan, Abhilash
author_facet Wahd, Assefa
Jaremko, Jacob
Hareendranathan, Abhilash
contents In-context learning (ICL) enables generalization to new tasks with minimal labeled data. However, mainstream ICL approaches rely on a gridding strategy, which lacks the flexibility required for vision applications. We introduce Temporal, a time-contrastive self-supervised objective that pretrains a prompt retriever for visual ICL, and formulate ICL as a video object segmentation (VOS) task. Temporal addresses key limitations of grid-based methods that restrict the number and resolution of context images. By reframing ICL as a VOS problem, our approach supports a variable number of context images while preserving their full resolution. To address the challenge of selecting optimal context sets for queries, we pretrain a prompt retriever on videos via self-supervised learning, where adjacent frames serve as positives and distant frames as negatives. For image segmentation, the prompt retriever selects relevant sequences that, when combined with the query, form coherent videos for VOS processing. For video segmentation, it identifies keyframes, predicts their masks using our ICL pipeline, and propagates them throughout the sequence. When evaluated on MICCAI FLARE 2022, our method achieves substantial improvements over baselines: 90.95% Dice score for image segmentation (10.64% improvement) and 92.45% Dice for video segmentation (14.88% improvement).
format Preprint
id arxiv_https___arxiv_org_abs_2506_17837
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Time-Contrastive Pretraining for In-Context Image and Video Segmentation
Wahd, Assefa
Jaremko, Jacob
Hareendranathan, Abhilash
Computer Vision and Pattern Recognition
In-context learning (ICL) enables generalization to new tasks with minimal labeled data. However, mainstream ICL approaches rely on a gridding strategy, which lacks the flexibility required for vision applications. We introduce Temporal, a time-contrastive self-supervised objective that pretrains a prompt retriever for visual ICL, and formulate ICL as a video object segmentation (VOS) task. Temporal addresses key limitations of grid-based methods that restrict the number and resolution of context images. By reframing ICL as a VOS problem, our approach supports a variable number of context images while preserving their full resolution. To address the challenge of selecting optimal context sets for queries, we pretrain a prompt retriever on videos via self-supervised learning, where adjacent frames serve as positives and distant frames as negatives. For image segmentation, the prompt retriever selects relevant sequences that, when combined with the query, form coherent videos for VOS processing. For video segmentation, it identifies keyframes, predicts their masks using our ICL pipeline, and propagates them throughout the sequence. When evaluated on MICCAI FLARE 2022, our method achieves substantial improvements over baselines: 90.95% Dice score for image segmentation (10.64% improvement) and 92.45% Dice for video segmentation (14.88% improvement).
title Time-Contrastive Pretraining for In-Context Image and Video Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2506.17837