← Search

Gao Huang

121 accepted papers

2026

Linearizing Vision Transformer with Test-Time Training

ICML 2026poster

While linear-complexity attention mechanisms offer a promising alternative to Softmax attention for overcoming the quadratic bottleneck, training such models from scratch remains prohibitively expensive. Inheriting weights from pretrained Transformers provides an appealing shortcut, yet the fundamen…

Cited by 0SourceScholar
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…

2026

MemoryVLA: Perceptual-Cognitive Memory in Vision-Language-Action Models for Robotic Manipulation

ICLR 2026poster

Temporal context is essential for robotic manipulation because such tasks are inherently non-Markovian, yet mainstream VLA models typically overlook it and struggle with long-horizon, temporally dependent tasks. Cognitive science suggests that humans rely on working memory to buffer short-lived repr…

Cited by 0SourcecodeScholar
2026

SenseSearch: Empowering Vision-Language Models with High-Resolution Agentic Search-Reasoning via Reinforcement Learning

CVPR 2026

Vision-Language Models (VLMs) are limited by static knowledge and insufficient fine-grained visual analysis, hindering their performance on knowledge-intensive and visually complex tasks. While recent research has explored VLMs that employ external tools like search or cropping to enhance model perf

Cited by 0SourcecodeScholar
2026

SiameseNorm: Breaking the Barrier to Reconciling Pre/Post-Norm

ICML 2026poster

Modern Transformers predominantly adopt the Pre-Norm paradigm for its optimization stability, foregoing the superior potential of the unstable Post-Norm architecture. Prior attempts to combine their strengths typically lead to a stability-performance trade-off. We attribute this phenomenon to a stru…

Cited by 0SourceScholar
2026

SpatialActor: Exploring Disentangled Spatial Representations for Robust Robotic Manipulation

AAAI 2026technical

Robotic manipulation requires precise spatial understanding to interact with objects in the real world. Point-based methods suffer from sparse sampling, leading to the loss of fine-grained semantics. Image-based methods typically feed RGB and depth into 2D backbones pre-trained on 3D auxiliary tasks

Cited by 0SourcePDFScholar
2026

The Flexibility Trap: Rethinking the Value of Arbitrary Order in Diffusion Language Models

ICML 2026oral

Diffusion Large Language Models (dLLMs) break the rigid left-to-right constraint of traditional LLMs, enabling token generation in arbitrary orders. Intuitively, this flexibility implies a solution space that strictly supersets the fixed autoregressive trajectory, theoretically unlocking superior re…

Cited by 0SourceScholar
2026

UltraHiT: A Hierarchical Transformer Architecture for Generalizable Internal Carotid Artery Robotic Ultrasonography

ICRA 2026poster

Carotid ultrasound is crucial for the assessment of cerebrovascular health, particularly the internal carotid artery (ICA). While previous research has explored automating carotid ultrasound, none has tackled the challenging ICA. This is primarily due to its deep location, tortuous course, and signi…

2025

4D LangSplat: 4D Language Gaussian Splatting via Multimodal Large Language Models

CVPR 2025poster

Learning 4D language fields to enable time-sensitive, open-ended language queries in dynamic scenes is essential for many real-world applications. While LangSplat successfully grounds CLIP features into 3D Gaussian representations, achieving precision and efficiency in 3D static scenes, it lacks the…

2025

ART: Anonymous Region Transformer for Variable Multi-Layer Transparent Image Generation

CVPR 2025poster

Multi-layer image generation is a fundamental task that enables users to isolate, select, and edit specific image layers, thereby revolutionizing interactions with generative models. In this paper, we introduce the Anonymous Region Transformer (ART), which facilitates the direct generation of variab…

Cited by 4SourcePDFScholar
2025

Absolute Zero: Reinforced Self-play Reasoning with Zero Data

NeurIPS 2025spotlight

Reinforcement learning with verifiable rewards (RLVR) has shown promise in enhancing the reasoning capabilities of large language models by learning directly from rule-based outcome rewards. Recent RLVR works that operate under the zero setting avoid supervision in labeling the reasoning process, bu…

Cited by 0SourceScholar
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

CODA: Repurposing Continuous VAEs for Discrete Tokenization

ICCV 2025poster

Discrete visual tokenizers transform images into a sequence of tokens, enabling token-based visual generation akin to language models. However, this process is inherently challenging, as it requires both compressing visual signals into a compact representation and discretizing them into a fixed set…

Cited by 0SourcePDFScholar
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

DTOS: Dynamic Time Object Sensing with Large Multimodal Model

CVPR 2025poster

Existing multimodal large language models (MLLMs) face significant challenges in Referring Video Object Segmentation(RVOS). We identify three critical challenges: (C1) insufficient quantitative representation of textual numerical data, (C2) repetitive and degraded response templates for spatiotempor…

2025

DenseGrounding: Improving Dense Language-Vision Semantics for Ego-centric 3D Visual Grounding

ICLR 2025poster

Enabling intelligent agents to comprehend and interact with 3D environments through natural language is crucial for advancing robotics and human-computer interaction. A fundamental task in this field is ego-centric 3D visual grounding, where agents locate target objects in real-world 3D spaces based…

Cited by 0SourcePDFScholar
2025

DiveR-CT: Diversity-enhanced Red Teaming Large Language Model Assistants with Relaxing Constraints

AAAI 2025technical

Recent advances in large language model assistants have made them indispensable, raising significant concerns over managing their safety. Automated red teaming offers a promising alternative to the labor-intensive and error-prone manual probing for vulnerabilities, providing more consistent and scal…

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

DyMoDreamer: World Modeling with Dynamic Modulation

NeurIPS 2025poster

A critical bottleneck in deep reinforcement learning (DRL) is sample inefficiency, as training high-performance agents often demands extensive environmental interactions. Model-based reinforcement learning (MBRL) mitigates this by building world models that simulate environmental dynamics and genera…

Cited by 0SourceScholar
2025

Dynamic Diffusion Transformer

ICLR 2025poster

Diffusion Transformer (DiT), an emerging diffusion model for image generation, has demonstrated superior performance but suffers from substantial computational costs. Our investigations reveal that these costs stem from the static inference paradigm, which inevitably introduces redundant computation…

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

How Far Is Video Generation from World Model: A Physical Law Perspective

ICML 2025poster

Scaling video generation models is believed to be promising in building world models that adhere to fundamental physical laws. However, whether these models can discover physical laws purely from vision can be questioned. A world model learning the true law should give predictions robust to nuances…

Cited by 35SourcePDFScholar
2025

IMG: Calibrating Diffusion Models via Implicit Multimodal Guidance

ICCV 2025poster

Ensuring precise multimodal alignment between diffusion-generated images and input prompts has been a long-standing challenge. Earlier works finetune diffusion weight using high-quality preference data, which tends to be limited and difficult to scale up. Recent editing-based methods further refine…

2025

Linear Differential Vision Transformer: Learning Visual Contrasts via Pairwise Differentials

NeurIPS 2025poster

Vision Transformers (ViTs) have become a universal backbone for both image recognition and image generation. Yet their Multi–Head Self–Attention (MHSA) layer still performs a quadratic query–key interaction for \emph{every} token pair, spending the bulk of computation on visually weak or redundant…

Cited by 0SourcecodeScholar
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…

2025

ProxyTransformation: Preshaping Point Cloud Manifold With Proxy Attention For 3D Visual Grounding

CVPR 2025poster

Embodied intelligence requires agents to interact with 3D environments in real time based on language instructions. A foundational task in this domain is ego-centric 3D visual grounding. However, the point clouds rendered from RGB-D images retain a large amount of redundant background data and inher…

Cited by 0SourcePDFScholar
2025

Towards Understanding Text Hallucination of Diffusion Models via Local Generation Bias

ICLR 2025poster

Score-based diffusion models have achieved incredible performance in generating realistic images, audio, and video data. While these models produce high-quality samples with impressive details, they often introduce unrealistic artifacts, such as distorted fingers or hallucinated texts with no meanin…

Cited by 0SourcePDFScholar
2025

UltraDP: Generalizable Carotid Ultrasound Scanning with Force-Aware Diffusion Policy

IROS 2025

Ultrasound scanning is a critical imaging technique for real-time, non-invasive diagnostics. However, variations in patient anatomy and complex human-in-the-loop interactions pose significant challenges for autonomous robotic scanning. Existing ultrasound scanning robots are commonly limited to rela

Cited by 3SourceScholar
2025

Video Perception Models for 3D Scene Synthesis

NeurIPS 2025poster

Automating the expert-dependent and labor-intensive task of 3D scene synthesis would significantly benefit fields such as architectural design, robotics simulation, and virtual reality. Recent approaches to 3D scene synthesis often rely on the commonsense reasoning of large language models (LLMs) or…

Cited by 0SourceScholar
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

ADDP: Learning General Representations for Image Recognition and Generation with Alternating Denoising Diffusion Process

ICLR 2024poster

Image recognition and generation have long been developed independently of each other. With the recent trend towards general-purpose representation learning, the development of general representations for both recognition and generation tasks is also promoted. However, preliminary attempts mainly fo…

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

COVE: Unleashing the Diffusion Feature Correspondence for Consistent Video Editing

NeurIPS 2024poster

Video editing is an emerging task, in which most current methods adopt the pre-trained text-to-image (T2I) diffusion model to edit the source video in a zero-shot manner. Despite extensive efforts, maintaining the temporal consistency of edited videos remains challenging due to the lack of temporal…

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

Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation

NeurIPS 2024poster

Existing parameter-efficient fine-tuning (PEFT) methods have achieved significant success on vision transformers (ViTs) adaptation by improving parameter efficiency. However, the exploration of enhancing inference efficiency during adaptation remains underexplored. This limits the broader applicatio…

2024

ENAT: Rethinking Spatial-temporal Interactions in Token-based Image Synthesis

NeurIPS 2024poster

Recently, token-based generation approaches have demonstrated their effectiveness in synthesizing visual content. As a representative example, non-autoregressive Transformers (NATs) can generate decent-quality images in just a few steps. NATs perform generation in a progressive manner, where the lat…

2024

ExpeL: LLM Agents Are Experiential Learners

AAAI 2024technical

The recent surge in research interest in applying large language models (LLMs) to decision-making tasks has flourished by leveraging the extensive world knowledge embedded in LLMs. While there is a growing demand to tailor LLMs for custom decision-making tasks, finetuning them for specific tasks is…

2024

Exploring Temporal Feature Correlation for Efficient and Stable Video Semantic Segmentation

AAAI 2024technical

This paper tackles the problem of efficient and stable video semantic segmentation. While stability has been under-explored, prevalent work in efficient video semantic segmentation uses the keyframe paradigm. They efficiently process videos by only recomputing the low-level features and reusing high…

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

Glyph-ByT5: A Customized Text Encoder for Accurate Visual Text Rendering

ECCV 2024poster

"Visual text rendering poses a fundamental challenge for contemporary text-to-image generation models, with the core problem lying in text encoder deficiencies. To achieve accurate text rendering, we identify two crucial requirements for text encoders: character awareness and alignment with glyphs.…

2024

Learning 1D Causal Visual Representation with De-focus Attention Networks

NeurIPS 2024poster

Modality differences have led to the development of heterogeneous architectures for vision and language models. While images typically require 2D non-causal modeling, texts utilize 1D causal modeling. This distinction poses significant challenges in constructing unified multi-modal models. This pape…

2024

Mask Grounding for Referring Image Segmentation

CVPR 2024poster

Referring Image Segmentation (RIS) is a challenging task that requires an algorithm to segment objects referred by free-form language expressions. Despite significant progress in recent years most state-of-the-art (SOTA) methods still suffer from considerable language-image modality gap at the pixel…

2024

Prompt-Free Diffusion: Taking "Text" out of Text-to-Image Diffusion Models

CVPR 2024poster

Text-to-image (T2I) research has grown explosively in the past year owing to the large-scale pre-trained diffusion models and many emerging personalization and editing approaches. Yet one pain point persists: the text prompt engineering and searching high-quality text prompts for customized results…

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

SimPro: A Simple Probabilistic Framework Towards Realistic Long-Tailed Semi-Supervised Learning

ICML 2024poster

Recent advancements in semi-supervised learning have focused on a more realistic yet challenging task: addressing imbalances in labeled data while the class distribution of unlabeled data remains both unknown and potentially mismatched. Current approaches in this sphere often presuppose rigid assump…

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

Training an Open-Vocabulary Monocular 3D Detection Model without 3D Data

NeurIPS 2024poster

Open-vocabulary 3D object detection has recently attracted considerable attention due to its broad applications in autonomous driving and robotics, which aims to effectively recognize novel classes in previously unseen domains. However, existing point cloud-based open-vocabulary 3D detection models…

Cited by 4SourcePDFScholar
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

BEVFormer v2: Adapting Modern Image Backbones to Bird's-Eye-View Recognition via Perspective Supervision

CVPR 2023highlight

We present a novel bird's-eye-view (BEV) detector with perspective supervision, which converges faster and better suits modern image backbones. Existing state-of-the-art BEV detectors are often tied to certain depth pre-trained backbones like VoVNet, hindering the synergy between booming image backb…

Cited by 316SourcePDFScholar
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

Borrowing Knowledge From Pre-trained Language Model: A New Data-efficient Visual Learning Paradigm

ICCV 2023poster

The development of vision models for real-world applications is hindered by the challenge of annotated data scarcity, which has necessitated the adoption of data-efficient visual learning techniques such as semi-supervised learning. Unfortunately, the prevalent cross-entropy supervision is limited b…

Cited by 8PDFcodeScholar
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

Deep Incubation: Training Large Models by Divide-and-Conquering

ICCV 2023poster

Recent years have witnessed a remarkable success of large deep learning models. However, training these models is challenging due to high computational costs, painfully slow convergence, and overfitting issues. In this paper, we present Deep Incubation, a novel approach that enables the efficient an…

Cited by 11PDFcodeScholar
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

Learning to Estimate 3-D States of Deformable Linear Objects from Single-Frame Occluded Point Clouds

ICRA 2023poster

Accurately and robustly estimating the state of deformable linear objects (DLOs), such as ropes and wires, is crucial for DLO manipulation and other applications. However, it remains a challenging open issue due to the high dimensionality of the state space, frequent occlusions, and noises. This pap…

Cited by 19SourceScholar
2023

Rank-DETR for High Quality Object Detection

NeurIPS 2023poster

Modern detection transformers (DETRs) use a set of object queries to predict a list of bounding boxes, sort them by their classification confidence scores, and select the top-ranked predictions as the final detection results for the given input image. A highly performant object detector requires acc…

2023

STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learning

NeurIPS 2023poster

Recently, model-based reinforcement learning algorithms have demonstrated remarkable efficacy in visual input environments. These approaches begin by constructing a parameterized simulation world model of the real environment through self-supervised learning. By leveraging the imagination of the wo…

2023

Siamese Image Modeling for Self-Supervised Vision Representation Learning

CVPR 2023poster

Self-supervised learning (SSL) has delivered superior performance on a variety of downstream vision tasks. Two main-stream SSL frameworks have been proposed, i.e., Instance Discrimination (ID) and Masked Image Modeling (MIM). ID pulls together representations from different views of the same image,…

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

Towards All-in-One Pre-Training via Maximizing Multi-Modal Mutual Information

CVPR 2023poster

To effectively exploit the potential of large-scale models, various pre-training strategies supported by massive data from different sources are proposed, including supervised pre-training, weakly-supervised pre-training, and self-supervised pre-training. It has been proved that combining multiple p…

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

Value-Consistent Representation Learning for Data-Efficient Reinforcement Learning

AAAI 2023technical

Deep reinforcement learning (RL) algorithms suffer severe performance degradation when the interaction data is scarce, which limits their real-world application. Recently, visual representation learning has been shown to be effective and promising for boosting sample efficiency in RL. These methods…

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

A Mixture Of Surprises for Unsupervised Reinforcement Learning

NeurIPS 2022accept

Unsupervised reinforcement learning aims at learning a generalist policy in a reward-free manner for fast adaptation to downstream tasks. Most of the existing methods propose to provide an intrinsic reward based on surprise. Maximizing or minimizing surprise drives the agent to either explore or gai…

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

AdaFocus V2: End-to-End Training of Spatial Dynamic Networks for Video Recognition

CVPR 2022poster

Recent works have shown that the computational efficiency of video recognition can be significantly improved by reducing the spatial redundancy. As a representative work, the adaptive focus method (AdaFocus) has achieved a favorable trade-off between accuracy and inference speed by dynamically ident…

Cited by 63PDFcodeScholar
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

Assessing a Single Image in Reference-Guided Image Synthesis

AAAI 2022technical

Assessing the performance of Generative Adversarial Networks (GANs) has been an important topic due to its practical significance. Although several evaluation metrics have been proposed, they generally assess the quality of the whole generated image distribution. For Reference-guided Image Synthesis…

Cited by 15SourcePDFScholar
2022

AutoLoss-Zero: Searching Loss Functions From Scratch for Generic Tasks

CVPR 2022poster

Significant progress has been achieved in automating the design of various components in deep networks. However, the automatic design of loss functions for generic tasks with various evaluation metrics remains under-investigated. Previous works on handcrafting loss functions heavily rely on human ex…

Cited by 43PDFScholar
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

DiSparse: Disentangled Sparsification for Multitask Model Compression

CVPR 2022poster

Despite the popularity of Model Compression and Multitask Learning, how to effectively compress a multitask model has been less thoroughly analyzed due to the challenging entanglement of tasks in the parameter space. In this paper, we propose DiSparse, a simple, effective, and first-of-its-kind mult…

Cited by 23PDFcodeScholar
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

Provable General Function Class Representation Learning in Multitask Bandits and MDP

NeurIPS 2022accept

While multitask representation learning has become a popular approach in reinforcement learning (RL) to boost the sample efficiency, the theoretical understanding of why and how it works is still limited. Most previous analytical works could only assume that the representation function is already kn…

Cited by 10SourcePDFScholar
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

Auto Seg-Loss: Searching Metric Surrogates for Semantic Segmentation

ICLR 2021poster

Designing proper loss functions is essential in training deep networks. Especially in the field of semantic segmentation, various evaluation metrics have been proposed for diverse scenarios. Despite the success of the widely adopted cross-entropy loss and its variants, the mis-alignment between the…

2021

Believe What You See: Implicit Constraint Approach for Offline Multi-Agent Reinforcement Learning

NeurIPS 2021spotlight

Learning from datasets without interaction with environments (Offline Learning) is an essential step to apply Reinforcement Learning (RL) algorithms in real-world scenarios. However, compared with the single-agent counterpart, offline multi-agent RL introduces more agents with the larger state and a…

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

Evolving Attention with Residual Convolutions

ICML 2021spotlight

Transformer is a ubiquitous model for natural language processing and has attracted wide attentions in computer vision. The attention maps are indispensable for a transformer model to encode the dependencies among input tokens. However, they are learned independently in each layer and sometimes fail…

2021

Frequency Domain Image Translation: More Photo-Realistic, Better Identity-Preserving

ICCV 2021poster

Image-to-image translation has been revolutionized with GAN-based methods. However, existing methods lack the ability to preserve the identity of the source domain. As a result, synthesized images can often over-adapt to the reference domain, losing important structural characteristics and suffering…

Cited by 106PDFcodeScholar
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

Searching Parameterized AP Loss for Object Detection

NeurIPS 2021poster

Loss functions play an important role in training deep-network-based object detectors. The most widely used evaluation metric for object detection is Average Precision (AP), which captures the performance of localization and classification sub-tasks simultaneously. However, due to the non-differenti…

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
2020

Spatially Adaptive Inference with Stochastic Feature Sampling and Interpolation

ECCV 2020poster

In the feature maps of CNNs, there commonly exists considerable spatial redundancy that leads to much repetitive processing. Towards reducing this superfluous computation, we propose to compute features only at sparsely sampled locations, which are probabilistically chosen according to activation re…

2019

Anytime Stereo Image Depth Estimation on Mobile Devices

ICRA 2019poster

Many applications of stereo depth estimation in robotics require the generation of accurate disparity maps in real time under significant computational constraints. Current state-of-the-art algorithms force a choice between either generating accurate mappings at a slow pace, or quickly generating in…

Cited by 265SourcecodeScholar
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…

2019

Rethinking the Value of Network Pruning

ICLR 2019poster

Network pruning is widely used for reducing the heavy inference cost of deep models in low-resource settings. A typical pruning algorithm is a three-stage pipeline, i.e., training (a large model), pruning and fine-tuning. During pruning, according to a certain criterion, redundant weights are pruned…

2018

CondenseNet: An Efficient DenseNet Using Learned Group Convolutions

CVPR 2018poster

Deep neural networks are increasingly used on mobile devices, where computational resources are limited. In this paper we develop CondenseNet, a novel network architecture with unprecedented efficiency. It combines dense connectivity with a novel module called learned group convolution. The dense co…

2018

Multi-Scale Dense Networks for Resource Efficient Image Classification

ICLR 2018oral

In this paper we investigate image classification with computational resource limits at test time. Two such settings are: 1. anytime classification, where the network’s prediction for a test example is progressively updated, facilitating the output of a prediction at any time; and 2. budgeted batch…

Cited by 941SourcePDFScholar
2018

Resource Aware Person Re-Identification Across Multiple Resolutions

CVPR 2018poster

Not all people are equally easy to identify: color statistics might be enough for some cases while others might require careful reasoning about high- and low-level details. However, prevailing person re-identification(re-ID) methods use one-size-fits-all high-level embeddings from deep convolutional…

2017

Learning Efficient Convolutional Networks Through Network Slimming

ICCV 2017poster

The deployment of deep convolutional neural networks (CNNs) in many real world applications is largely hindered by their high computational cost. In this paper, we propose a novel learning scheme for CNNs to simultaneously 1) reduce the model size; 2) decrease the run-time memory footprint; and 3) l…

Cited by 3354PDFcodeScholar
2017

Snapshot Ensembles: Train 1, Get M for Free

ICLR 2017poster

Ensembles of neural networks are known to be much more robust and accurate than individual networks. However, training multiple deep networks for model averaging is computationally expensive. In this paper, we propose a method to obtain the seemingly contradictory goal of ensembling multiple neural…

Cited by 1227SourcecodeScholar
2016

Supervised Word Mover's Distance

NeurIPS 2016oral

Accurately measuring the similarity between text documents lies at the core of many real world applications of machine learning. These include web-search ranking, document recommendation, multi-lingual document matching, and article categorization. Recently, a new document metric, the word mover's d…