Paper Reading AI Learner

Mitigating Surgical Data Imbalance with Dual-Prediction Video Diffusion Model

2025-10-07 20:29:27
Danush Kumar Venkatesh, Adam Schmidt, Muhammad Abdullah Jamal, Omid Mohareri

Abstract

Surgical video datasets are essential for scene understanding, enabling procedural modeling and intra-operative support. However, these datasets are often heavily imbalanced, with rare actions and tools under-represented, which limits the robustness of downstream models. We address this challenge with $SurgiFlowVid$, a sparse and controllable video diffusion framework for generating surgical videos of under-represented classes. Our approach introduces a dual-prediction diffusion module that jointly denoises RGB frames and optical flow, providing temporal inductive biases to improve motion modeling from limited samples. In addition, a sparse visual encoder conditions the generation process on lightweight signals (e.g., sparse segmentation masks or RGB frames), enabling controllability without dense annotations. We validate our approach on three surgical datasets across tasks including action recognition, tool presence detection, and laparoscope motion prediction. Synthetic data generated by our method yields consistent gains of 10-20% over competitive baselines, establishing $SurgiFlowVid$ as a promising strategy to mitigate data imbalance and advance surgical video understanding methods.

Abstract (translated)

外科手术视频数据集对于场景理解至关重要,可以促进程序建模和术中支持。然而,这些数据集往往不平衡,罕见的动作和工具被严重低估,这限制了下游模型的鲁棒性。我们通过引入$SurgiFlowVid$来解决这一挑战,这是一个稀疏且可控的视频扩散框架,用于生成代表性不足类别的手术视频。我们的方法介绍了一个双预测扩散模块,该模块同时对RGB帧和光流进行去噪处理,提供时间归纳偏差以从有限样本中改进运动建模。此外,一个稀疏视觉编码器根据轻量级信号(例如稀疏分割掩码或RGB帧)来调节生成过程,从而在没有密集注释的情况下实现可控性。 我们在三个手术数据集上验证了我们的方法,涵盖了包括动作识别、工具存在检测和腹腔镜运动预测在内的任务。我们通过实验表明,由我们的方法生成的合成数据相对于竞争基线带来了10-20%的一致性能提升,从而将$SurgiFlowVid$确立为一种有前景的方法来缓解数据不平衡并推进手术视频理解技术的进步。

URL

https://arxiv.org/abs/2510.07345

PDF

https://arxiv.org/pdf/2510.07345.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot