← Search

Shiji Song

57 accepted papers

2026

From Swept Contact to Pose: Probe-Aware Registration Via Complementary-Shape Docking

ICRA 2026poster

Accurate registration between a prior model and the real scene is essential for high-precision robotic manipulation, yet optical methods suffer from long calibration chains, line-of-sight constraints, and fabrication errors. We propose a calibration-free alternative that reformulates contact registr…

2026

MOVE: A Simple Motion-Based Data Collection Paradigm for Spatial Generalization in Robotic Manipulation

ICRA 2026poster

Imitation learning method has shown immense promise for robotic manipulation, yet its practical deployment is fundamentally constrained by the data scarcity. Despite prior work on collecting large-scale datasets, there still remains a significant gap to robust spatial generalization. We identify a k…

2025

Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning

NeurIPS 2025poster

Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a powerful approach to enhancing the reasoning capabilities of Large Language Models (LLMs), yet its underlying mechanisms remain insufficiently understood. In this work, we undertake a pioneering exploration of RLVR through the no…

Cited by 0SourceScholar
2025

CheXWorld: Exploring Image World Modeling for Radiograph Representation Learning

CVPR 2025poster

Humans can develop internal world models that encode common sense knowledge, telling them how the world works and predicting the consequences of their actions. This concept has emerged as a promising direction for establishing general-purpose machine-learning models in recent preliminary works, e.g.…

2025

Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?

NeurIPS 2025oral

Reinforcement Learning with Verifiable Rewards (RLVR) has recently demonstrated notable success in enhancing the reasoning performance of large language models (LLMs), particularly in mathematics and programming tasks. It is widely believed that, similar to how traditional RL helps agents to explor…

Cited by 0SourceScholar
2025

EchoWorld: Learning Motion-Aware World Models for Echocardiography Probe Guidance

CVPR 2025poster

Echocardiography is crucial for cardiovascular disease detection but relies heavily on experienced sonographers. Echocardiography probe guidance systems, which provide real-time movement instructions for acquiring standard plane images, offer a promising solution for AI-assisted or fully autonomous…

2025

Everything to the Synthetic: Diffusion-driven Test-time Adaptation via Synthetic-Domain Alignment

CVPR 2025poster

Test-time adaptation (TTA) aims to improve the performance of source-domain pre-trained models on previously unseen, shifted target domains. Traditional TTA methods primarily adapt model weights based on target data streams, making model performance sensitive to the amount and order of target data.…

2025

Model Surgery: Modulating LLM’s Behavior Via Simple Parameter Editing

NAACL 2025long

Large Language Models (LLMs) have demonstrated great potential as generalist assistants, showcasing powerful task understanding and problem-solving capabilities. To deploy LLMs as AI assistants, it is crucial that these models exhibit desirable behavioral traits, such as non-toxicity and resilience…

2024

A Reinforcement-Learning-Based Multiple-Column Selection Strategy for Column Generation

AAAI 2024technical

Column generation (CG) is one of the most successful approaches for solving large-scale linear programming (LP) problems. Given an LP with a prohibitively large number of variables (i.e., columns), the idea of CG is to explicitly consider only a subset of columns and iteratively add potential column…

Cited by 5SourcePDFScholar
2024

A Unified Interaction Control Framework for Safe Robotic Ultrasound Scanning with Human-Intention-Aware Compliance

IROS 2024

The ultrasound scanning robot operates in environments where frequent human-robot interactions occur. Most existing control methods for ultrasound scanning address only one specific interaction situation or implement hard switches between controllers for different situations, which compromises both

Cited by 6SourceScholar
2024

Agent Attention: On the Integration of Softmax and Linear Attention

ECCV 2024poster

"The attention module is the key component in Transformers. While the global attention mechanism offers high expressiveness, its excessive computational cost restricts its applicability in various scenarios. In this paper, we propose a novel attention paradigm, Agent Attention, to strike a favorable…

2024

Boosting LLM Agents with Recursive Contemplation for Effective Deception Handling

ACL 2024findings

Recent advances in large language models (LLMs) have led to significant success in using LLMs as agents. Nevertheless, a common assumption that LLMs always process honest information neglects the widespread deceptive or misleading content in human and AI-generated material. This oversight might expo…

2024

Bridging the Divide: Reconsidering Softmax and Linear Attention

NeurIPS 2024poster

Widely adopted in modern Vision Transformer designs, Softmax attention can effectively capture long-range visual information; however, it incurs excessive computational cost when dealing with high-resolution inputs. In contrast, linear attention naturally enjoys linear complexity and has great poten…

2024

DeeR-VLA: Dynamic Inference of Multimodal Large Language Models for Efficient Robot Execution

NeurIPS 2024poster

Multimodal Large Language Models (MLLMs) have demonstrated remarkable comprehension and reasoning capabilities with complex language and visual data. These advances have spurred the vision of establishing a generalist robotic MLLM proficient in understanding complex human instructions and accomplish…

2024

Demystify Mamba in Vision: A Linear Attention Perspective

NeurIPS 2024poster

Mamba is an effective state space model with linear computation complexity. It has recently shown impressive efficiency in dealing with high-resolution inputs across various vision tasks. In this paper, we reveal that the powerful Mamba model shares surprising similarities with linear attention Tran…

2024

DyFADet: Dynamic Feature Aggregation for Temporal Action Detection

ECCV 2024poster

"Recent proposed neural network-based Temporal Action Detection (TAD) models are inherently limited to extracting the discriminative representations and modeling action instances with various lengths from complex scenes by shared-weights detection heads. Inspired by the successes in dynamic neural n…

2024

Efficient Diffusion Transformer with Step-wise Dynamic Attention Mediators

ECCV 2024poster

"This paper identifies significant redundancy in the query-key interactions within self-attention mechanisms of diffusion transformer models, particularly during the early stages of denoising diffusion steps. In response to this observation, we present a novel diffusion transformer framework incorpo…

2024

GSVA: Generalized Segmentation via Multimodal Large Language Models

CVPR 2024poster

Generalized Referring Expression Segmentation (GRES) extends the scope of classic RES to refer to multiple objects in one expression or identify the empty targets absent in the image. GRES poses challenges in modeling the complex spatial relationships of the instances in the image and identifying no…

2024

In-Hand Following of Deformable Linear Objects Using Dexterous Fingers with Tactile Sensing

IROS 2024

Most research on deformable linear object (DLO) manipulation assumes rigid grasping. However, beyond rigid grasping and re-grasping, in-hand following is also an essential skill that humans use to dexterously manipulate DLOs, which requires continuously changing the grasp point by in-hand sliding wh

Cited by 13SourceScholar
2024

PsychoGAT: A Novel Psychological Measurement Paradigm through Interactive Fiction Games with LLM Agents

ACL 2024long

Psychological measurement is essential for mental health, self-understanding, and personal development. Traditional methods, such as self-report scales and psychologist interviews, often face challenges with engagement and accessibility. While game-based and LLM-based tools have been explored to imp…

Cited by 11SourcePDFScholar
2024

Revisiting Non-Autoregressive Transformers for Efficient Image Synthesis

CVPR 2024poster

The field of image synthesis is currently flourishing due to the advancements in diffusion models. While diffusion models have been successful their computational intensity has prompted the pursuit of more efficient alternatives. As a representative work non-autoregressive Transformers (NATs) have b…

2024

Segment3D: Learning Fine-Grained Class-Agnostic 3D Segmentation without Manual Labels

ECCV 2024poster

"Current 3D scene segmentation methods are heavily dependent on manually annotated 3D training datasets. Such manual annotations are labor-intensive, and often lack fine-grained details. Furthermore, models trained on this data typically struggle to recognize object classes beyond the annotated trai…

Cited by 34SourcePDFScholar
2024

Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models

CVPR 2024poster

Recently diffusion models have made remarkable progress in text-to-image (T2I) generation synthesizing images with high fidelity and diverse contents. Despite this advancement latent space smoothness within diffusion models remains largely unexplored. Smooth latent spaces ensure that a perturbation…

2024

Visual Attention Based Cognitive Human–Robot Collaboration for Pedicle Screw Placement in Robot-Assisted Orthopedic Surgery

IROS 2024poster

Current orthopedic robotic systems largely focus on navigation, aiding surgeons in positioning a guiding tube but still requiring manual drilling and screw placement. The automation of this task not only demands high precision and safety due to the intricate physical interactions between the surgica…

Cited by 1SourceScholar
2023

Adaptive Rotated Convolution for Rotated Object Detection

ICCV 2023poster

Rotated object detection aims to identify and locate objects in images with arbitrary orientation. In this scenario, the oriented directions of objects vary considerably across different images, while multiple orientations of objects exist within an image. This intrinsic characteristic makes it chal…

Cited by 117PDFcodeScholar
2023

Boosting Offline Reinforcement Learning with Action Preference Query

ICML 2023poster

Training practical agents usually involve offline and online reinforcement learning (RL) to balance the policy's performance and interaction costs. In particular, online fine-tuning has become a commonly used method to correct the erroneous estimates of out-of-distribution data learned in the offlin…

Cited by 11SourcePDFScholar
2023

Causal Intervention for Human Trajectory Prediction with Cross Attention Mechanism

AAAI 2023technical

Human trajectory Prediction (HTP) in complex social environments plays a crucial and fundamental role in artificial intelligence systems. Conventional methods make use of both history behaviors and social interactions to forecast future trajectories. However, we demonstrate that the social environme…

Cited by 18SourcePDFScholar
2023

Dynamic Perceiver for Efficient Visual Recognition

ICCV 2023poster

Early exiting has become a promising approach to im- proving the inference efficiency of deep networks. By structuring models with multiple classifiers (exits), predictions for "easy" samples can be generated at earlier exits, negating the need for executing deeper layers. Current multi-exit network…

Cited by 36PDFcodeScholar
2023

EfficientTrain: Exploring Generalized Curriculum Learning for Training Visual Backbones

ICCV 2023poster

The superior performance of modern deep networks usually comes with a costly training procedure. This paper presents a new curriculum learning approach for the efficient training of visual backbones (e.g., vision Transformers). Our work is inspired by the inherent learning dynamics of deep networks:…

Cited by 39PDFcodeScholar
2023

FLatten Transformer: Vision Transformer using Focused Linear Attention

ICCV 2023poster

The quadratic computation complexity of self-attention has been a persistent challenge when applying Transformer models to vision tasks. Linear attention, on the other hand, offers a much more efficient alternative with its linear complexity by approximating the Softmax operation through carefully d…

Cited by 239PDFcodeScholar
2023

Slide-Transformer: Hierarchical Vision Transformer With Local Self-Attention

CVPR 2023poster

Self-attention mechanism has been a key factor in the recent progress of Vision Transformer (ViT), which enables adaptive feature extraction from global contexts. However, existing self-attention methods either adopt sparse global attention or window attention to reduce the computation complexity, w…

2023

Train Once, Get a Family: State-Adaptive Balances for Offline-to-Online Reinforcement Learning

NeurIPS 2023spotlight

Offline-to-online reinforcement learning (RL) is a training paradigm that combines pre-training on a pre-collected dataset with fine-tuning in an online environment. However, the incorporation of online fine-tuning can intensify the well-known distributional shift problem. Existing solutions tackle…

2023

Understanding, Predicting and Better Resolving Q-Value Divergence in Offline-RL

NeurIPS 2023poster

The divergence of the Q-value estimation has been a prominent issue offline reinforcement learning (offline RL), where the agent has no access to real dynamics. Traditional beliefs attribute this instability to querying out-of-distribution actions when bootstrapping value targets. Though this issue…

2023

Zero-Shot Generative Model Adaptation via Image-Specific Prompt Learning

CVPR 2023poster

Recently, CLIP-guided image synthesis has shown appealing performance on adapting a pre-trained source-domain generator to an unseen target domain. It does not require any target-domain samples but only the textual domain labels. The training is highly efficient, e.g., a few minutes. However, existi…

2022

ActiveNeRF: Learning Where to See with Uncertainty Estimation

ECCV 2022poster

"Recently, Neural Radiance Fields (NeRF) has shown promising performances on reconstructing 3D scenes and synthesizing novel views from a sparse set of 2D images. Albeit effective, the performance of NeRF is highly influenced by the quality of training samples. With limited posed images from the sce…

2022

AdaFocusV3: On Unified Spatial-Temporal Dynamic Video Recognition

ECCV 2022poster

"Recent research has revealed that reducing the temporal and spatial redundancy are both effective approaches towards efficient video recognition, e.g., allocating the majority of computation to a task-relevant subset of frames or the most valuable image regions of each frame. However, in most exist…

Cited by 22SourcePDFScholar
2022

Contrastive Language-Image Pre-Training with Knowledge Graphs

NeurIPS 2022accept

Recent years have witnessed the fast development of large-scale pre-training frameworks that can extract multi-modal representations in a unified form and achieve promising performances when transferred to downstream tasks. Nevertheless, existing approaches mainly focus on pre-training with simple i…

Cited by 50SourcePDFScholar
2022

Efficient Knowledge Distillation from Model Checkpoints

NeurIPS 2022accept

Knowledge distillation is an effective approach to learn compact models (students) with the supervision of large and strong models (teachers). As empirically there exists a strong correlation between the performance of teacher and student models, it is commonly believed that a high performing teache…

2022

Exploring the Equivalence of Siamese Self-Supervised Learning via a Unified Gradient Framework

CVPR 2022poster

Self-supervised learning has shown its great potential to extract powerful visual representations without human annotations. Various works are proposed to deal with self-supervised learning from different perspectives: (1) contrastive learning methods (e.g., MoCo, SimCLR) utilize both positive and n…

Cited by 71PDFcodeScholar
2022

Latency-aware Spatial-wise Dynamic Networks

NeurIPS 2022accept

Spatial-wise dynamic convolution has become a promising approach to improving the inference efficiency of deep networks. By allocating more computation to the most informative pixels, such an adaptive inference paradigm reduces the spatial redundancy in image features and saves a considerable amount…

2022

Learning to Weight Samples for Dynamic Early-Exiting Networks

ECCV 2022poster

"Early exiting is an effective paradigm for improving the inference efficiency of deep networks. By constructing classifiers with varying resource demands (the exits), such networks allow easy samples to be output at early exits, removing the need for executing deeper layers. While existing works ma…

2022

On the Integration of Self-Attention and Convolution

CVPR 2022poster

Convolution and self-attention are two powerful techniques for representation learning, and they are usually considered as two peer approaches that are distinct from each other. In this paper, we show that there exists a strong underlying relation between them, in the sense that the bulk of computat…

Cited by 517PDFcodeScholar
2022

Pseudo-Q: Generating Pseudo Language Queries for Visual Grounding

CVPR 2022poster

Visual grounding, i.e., localizing objects in images according to natural language queries, is an important topic in visual language understanding. The most effective approaches for this task are based on deep learning, which generally require expensive manually labeled image-query or patch-query pa…

Cited by 73PDFcodeScholar
2021

Adaptive Focus for Efficient Video Recognition

ICCV 2021poster

In this paper, we explore the spatial redundancy in video recognition with the aim to improve the computational efficiency. It is observed that the most informative region in each frame of a video is usually a small image patch, which shifts smoothly across frames. Therefore, we model the patch loca…

Cited by 121PDFcodeScholar
2021

CondenseNet V2: Sparse Feature Reactivation for Deep Networks

CVPR 2021poster

Reusing features in deep networks through dense connectivity is an effective way to achieve high computational efficiency. The recent proposed CondenseNet has shown that this mechanism can be further improved if redundant features are removed. In this paper, we propose an alternative approach named…

Cited by 90PDFcodeScholar
2021

Not All Images are Worth 16x16 Words: Dynamic Transformers for Efficient Image Recognition

NeurIPS 2021poster

Vision Transformers (ViT) have achieved remarkable success in large-scale image recognition. They split every 2D image into a fixed number of patches, each of which is treated as a token. Generally, representing an image with more tokens would lead to higher prediction accuracy, while it also result…

2021

Revisiting Locally Supervised Learning: an Alternative to End-to-end Training

ICLR 2021poster

Due to the need to store the intermediate activations for back-propagation, end-to-end (E2E) training of deep networks usually suffers from high GPUs memory footprint. This paper aims to address this problem by revisiting the locally supervised learning, where a network is split into gradient-isolat…

2021

Towards Learning Spatially Discriminative Feature Representations

ICCV 2021poster

The backbone of traditional CNN classifier is generally considered as a feature extractor, followed by a linear layer which performs the classification. We propose a novel loss function, termed as CAM-loss, to constrain the embedded feature maps with the class activation maps (CAMs) which indicate t…

Cited by 28PDFScholar
2020

Glance and Focus: a Dynamic Approach to Reducing Spatial Redundancy in Image Classification

NeurIPS 2020poster

The accuracy of deep convolutional neural networks (CNNs) generally improves when fueled with high resolution images. However, this often comes at a high computational cost and high memory footprint. Inspired by the fact that not all regions in an image are task-relevant, we propose a novel framewor…

2020

Resolution Adaptive Networks for Efficient Inference

CVPR 2020poster

Adaptive inference is an effective mechanism to achieve a dynamic tradeoff between accuracy and computational cost in deep networks. Existing works mainly exploit architecture redundancy in network depth or width. In this paper, we focus on spatial redundancy of input samples and propose a novel Res…

Cited by 308PDFcodeScholar
2019

End-to-end sensorimotor control problems of AUVs with deep reinforcement learning

IROS 2019poster

This paper studies on sensorimotor control problems of Autonomous Underwater Vehicles (AUVs) using deep reinforcement learning. We design an end-to-end learning architecture mapping original sensor input to continuous control output without referring to the dynamics of vehicles. To avoid difficult a…

Cited by 25SourceScholar
2019

Implicit Semantic Data Augmentation for Deep Networks

NeurIPS 2019poster

In this paper, we propose a novel implicit semantic data augmentation (ISDA) approach to complement traditional augmentation techniques like flipping, translation or rotation. Our work is motivated by the intriguing property that deep networks are surprisingly good at linearizing features, such that…

2019

Regularized Anderson Acceleration for Off-Policy Deep Reinforcement Learning

NeurIPS 2019poster

Model-free deep reinforcement learning (RL) algorithms have been widely used for a range of complex control tasks. However, slow convergence and sample inefficiency remain challenging problems in RL, especially when handling continuous and high-dimensional state spaces. To tackle this problem, we pr…