← Search

Huan Wang

103 accepted papers

2026

Development of a Mixed-Control Ankle Assist Device with Sensor-Fusion-Based Phase Recognition for Walking Exercise Promotion

ICRA 2026poster

"Frail" elderly often experience walking impairments that limit independence and sustained physical activity. Although various assistive devices exist, many rely on single-mode control, limiting adaptability, responsiveness to gait variability, and voluntary motion. To improve, we developed a wearab…

Cited by 0Scholar
2026

EarlyTom: Early Token Compression Completes Fast Video Understanding

CVPR 2026

Video large language models (Video-LLMs) have demonstrated strong capabilities in video understanding tasks. However, their practical deployment is still hindered by the inefficiency introduced by processing massive amounts of visual tokens. Although recent approaches achieve extremely low token ret

Cited by 0SourceScholar
2026

Entropy-Based Block Pruning for Efficient Large Language Models

ICLR 2026poster

As large language models continue to scale, their growing computational and storage demands pose significant challenges for real-world deployment. In this work, we investigate redundancy within Transformer-based models and propose an entropy-based pruning strategy to enhance efficiency while maintai…

Cited by 0SourceScholar
2026

FedHPro: Federated Hyper-Prototype Learning via Gradient Matching

ICML 2026poster

Federated Learning (FL) enables collaborative training of distributed clients while protecting privacy. To enhance generalization capability in FL, prototype-based FL is in the spotlight, since shared global prototypes offer semantic anchors for aligning client-specific local prototypes. However, ex…

Cited by 0SourceScholar
2026

MergeMix: A Unified Augmentation Paradigm for Visual and Multi-Modal Understanding

ICLR 2026poster

Vision-language alignment in multi-modal large language models (MLLMs) relies on supervised fine-tuning (SFT) or reinforcement learning (RL). To align multi-modal large language models (MLLMs) in the post-training stage, supervised fine-tuning (SFT) is a stable choice but requires human annotations…

Cited by 0SourcecodeScholar
2026

OBS-Diff: Accurate Pruning For Diffusion Models in One-Shot

ICLR 2026poster

Large-scale text-to-image diffusion models, while powerful, suffer from prohibitive computational cost. Existing one-shot network pruning methods can hardly be directly applied to them due to the iterative denoising nature of diffusion models. To bridge the gap, this paper presents \textit{OBS-Diff}…

Cited by 0SourcecodeScholar
2026

OmniZip: Audio-Guided Dynamic Token Compression for Fast Omnimodal Large Language Models

CVPR 2026

Omnimodal large language models (OmniLLMs) have attracted increasing research attention of late towards unified audio-video understanding. However, the high computational cost of processing longer joint audio-video token sequences has become a key bottleneck. Existing token compression methods have

Cited by 0SourcecodeScholar
2026

Position: Vector Prompt Interfaces Should Be Exposed to Enable Customization of Large Language Models

ICML 2026poster

As large language models (LLMs) transition from research prototypes to real-world systems, customization has emerged as a central bottleneck. While text prompts can already customize LLM behavior, we argue that text-only prompting does not constitute a suitable control interface for scalable, stable…

Cited by 0SourceScholar
2026

Prism-MoE: Efficient Dense-to-MoE Conversion for Visual Autoregressive Generation

ICML 2026poster

Scaling up visual autoregressive models improves generation quality but incurs substantial inference costs. Mixture-of-Experts (MoE) architectures mitigate this issue through sparse activation and have proven effective in large language models. However, training MoE models from scratch remains prohi…

Cited by 0SourceScholar
2026

Probability-Driven Gating for Resilient Multi-Modal Tracking in Robotic Systems

ICRA 2026poster

The deployment of robots in unstructured environments demands perception systems that are both accurate and resilient. While RGB-Thermal (RGB-T) fusion is promising, current trackers often fail due to rigid, non-adaptive fusion strategies and underutilized cross-modal cues, compromising reliability …

Cited by 0Scholar
2026

ReasonMap: Towards Fine-Grained Visual Reasoning from Transit Maps

CVPR 2026

Multimodal large language models (MLLMs) have demonstrated significant progress in semantic scene understanding and text-image alignment, with reasoning variants enhancing performance on more complex tasks involving mathematics and logic. However, their proficiency in tasks requiring both fine-grain

Cited by 0SourcecodeScholar
2026

RewardMap: Tackling Sparse Rewards in Fine-grained Visual Reasoning via Multi-Stage Reinforcement Learning

ICLR 2026poster

Fine-grained visual reasoning remains a core challenge for multimodal large language models (MLLMs). The recently introduced ReasonMap highlights this gap by showing that even advanced MLLMs struggle with spatial reasoning in structured and information-rich settings such as transit maps, a task of c…

Cited by 0SourcecodeScholar
2026

Toward Multimodal Fake News Detection by Multi-perspective Rationale Generation and Verification

AAAI 2026technical

The rapid proliferation of social media platforms has led to a surge in multimodal fake news, where deceptive content often combines text and images to mislead audiences. Traditional unimodal detection methods struggle to address the complexity of such content, necessitating holistic multimodal appr

Cited by 0SourcePDFScholar
2026

Towards Zero-Shot Diabetic Retinopathy Grading: Learning Generalized Knowledge via Prompt-Driven Matching and Emulating

AAAI 2026technical

As one of the primary causes of visual impairment, Diabetic Retinopathy (DR) requires accurate and robust grading to facilitate timely diagnosis and intervention. Different from conventional DR grading methods that utilize single-view images, recent clinical studies have revealed that multi-view fun

Cited by 0SourcePDFScholar
2026

Webscale-RL: Automated Data Pipeline for Scaling RL Data to Pretraining Levels

ICLR 2026poster

Large Language Models (LLMs) have achieved remarkable success through imitation learning on vast text corpora, but this paradigm creates a training-generation gap and limits robust reasoning. Reinforcement learning (RL) offers a more data-efficient solution capable of bridging this gap, yet its appl…

Cited by 0SourcecodeScholar
2026

Zero-shot Recommendation: Towards Class Semantic Relation Learning for Inferring Labels of Unseen Micro-videos

AAAI 2026technical

Micro-video label prediction plays a pivotal role on contemporary video-sharing platforms, such as Kwai and Tiktok. The emergence of video content lacking labels presents a formidable challenge for conventional user interest prediction methods. This paper addresses the challenge of micro-video label

Cited by 0SourcePDFScholar
2025

A Reconfigurable Gripper Inspired by Elastic Belt for Versatile In-Hand Manipulations

RA-L 2025

Rigid and flexible grippers each have distinct advantages, yet their drawbacks are also evident. Combining the strengths of both types of grippers could significantly enhance the gripper's adaptability to various tasks. In this letter, we introduce a novel design of an elastic belt-based reconfigura

Cited by 0SourceScholar
2025

ABNet: Mitigating Sample Imbalance in Anomaly Detection Within Dynamic Graphs

IJCAI 2025

In dynamic graphs, detecting anomalous nodes faces challenges due to sample imbalance, stemming from the scarcity of anomalous samples and feature representation bias. Existing methods often use unsupervised or semi-supervised learning to extract anomalous samples from unlabeled data, but struggle t

Cited by 0SourcePDFScholar
2025

APIGen-MT: Agentic Pipeline for Multi-Turn Data Generation via Simulated Agent-Human Interplay

NeurIPS 2025poster

Training effective AI agents for multi-turn interactions requires high-quality data that captures realistic human-agent dynamics, yet such data is scarce and expensive to collect manually. We introduce APIGen-MT, a two-phase framework that generates verifiable and diverse multi-turn agent data. In t…

Cited by 0SourceScholar
2025

ActionStudio: A Lightweight Framework for Data and Training of Large Action Models

EMNLP 2025

Large Action models are essential for enabling autonomous agents to perform complex tasks. However, training such models remains challenging due to the diversity of agent environments and the complexity of noisy agentic data. Existing infrastructure offers limited support for scalable, agent-specifi

2025

Bioinspired Pad and Claw Composite Structure for Soft Robot Grasping and Crawling

RA-L 2025

The meat pads and claws serve as crucial functional components of cats' paws, which together enable the paws to exhibit remarkable dexterity. Inspired by extension and retraction of cat's claws, we have designed a composite structure combining pads and claws that can switch between pad state and cla

Cited by 1SourceScholar
2025

CRMArena: Understanding the Capacity of LLM Agents to Perform Professional CRM Tasks in Realistic Environments

NAACL 2025long

Customer Relationship Management (CRM) systems are vital for modern enterprises, providing a foundation for managing customer interactions and data. Integrating AI agents into CRM systems can automate routine processes and enhance personalized service. However, deploying and evaluating these agents…

2025

Collapsing Sequence-Level Data-Policy Coverage via Poisoning Attack in Offline Reinforcement Learning

UAI 2025

Offline reinforcement learning (RL) heavily relies on the coverage of pre-collected data over the target policy’s distribution. Existing studies aim to improve data-policy coverage to mitigate distributional shifts, but overlook security risks from insufficient coverage, and the single-step analysis

Cited by 0SourcePDFScholar
2025

Diffuse&Refine: Intrinsic Knowledge Generation and Aggregation for Incremental Object Detection

IJCAI 2025

Incremental Object Detection(IOD) targets at progressively extending capability of object detectors to recognize new classes. However, representation confusion between old and new classes leads to catastrophic forgetting. To alleviate this problem, we propose DiffKA, with intrinsic knowledge generat

Cited by 0SourcePDFScholar
2025

Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agents

ICLR 2025poster

Large language model (LLM) agents have shown great potential in solving real-world software engineering (SWE) problems. The most advanced open-source SWE agent can resolve over 27% of real GitHub issues in SWE-Bench Lite. However, these sophisticated agent frameworks exhibit varying strengths, excel…

Cited by 10SourcePDFScholar
2025

DyCoke: Dynamic Compression of Tokens for Fast Video Large Language Models

CVPR 2025poster

Video large language models (VLLMs) have significantly advanced recently in processing complex video content. Yet, their inference efficiency remains constrained because of the high computational cost stemming from the thousands of visual tokens generated from the video inputs. We empirically observ…

2025

FedDifRC: Unlocking the Potential of Text-to-Image Diffusion Models in Heterogeneous Federated Learning

ICCV 2025poster

Federated learning aims at training models collaboratively across participants while protecting privacy. However, one major challenge for this paradigm is the data heterogeneity issue, where biased data preferences across multiple clients, harming the model's convergence and performance. In this pap…

2025

FreqExit: Enabling Early-Exit Inference for Visual Autoregressive Models via Frequency-Aware Guidance

NeurIPS 2025poster

Visual AutoRegressive (VAR) modeling employs a next-scale decoding paradigm that progresses from coarse structures to fine details. While enhancing fidelity and scalability, this approach challenges two fundamental assumptions of conventional dynamic inference: semantic stability (intermediate outpu…

Cited by 0SourceScholar
2025

HoliTom: Holistic Token Merging for Fast Video Large Language Models

NeurIPS 2025poster

Video large language models (video LLMs) excel at video comprehension but face significant computational inefficiency due to redundant video tokens. Existing token pruning methods offer solutions. However, approaches operating within the LLM (inner-LLM pruning), such as FastV, incur intrinsic comput…

Cited by 0SourcecodeScholar
2025

LAM SIMULATOR: Advancing Data Generation for Large Action Model Training via Online Exploration and Trajectory Feedback

ACL 2025finding

Large Action Models (LAMs) for AI Agents offer incredible potential but face challenges due to the need for high-quality training data, especially for multi-steps tasks that involve planning, executing tool calls, and responding to feedback. To address these issues, we present LAM SIMULATOR, a compr…

Cited by 0SourcePDFScholar
2025

LATTE: Learning to Think with Vision Specialists

EMNLP 2025

While open-source vision-language models perform well on simple question-answering, they still struggle with complex questions that require both perceptual and reasoning capabilities. We propose LATTE, a family of vision-language models that have LeArned to Think wiTh vision spEcialists. By offloadi

2025

On-Device Diffusion Transformer Policy for Efficient Robot Manipulation

ICCV 2025poster

Diffusion Policies have significantly advanced robotic manipulation tasks via imitation learning, but their application on resource-constrained mobile platforms remains challenging due to computational inefficiency and extensive memory footprint. In this paper, we propose LightDP, a novel framework…

Cited by 0SourcePDFScholar
2025

PersonaBench: Evaluating AI Models on Understanding Personal Information through Accessing (Synthetic) Private User Data

ACL 2025finding

Personalization is essential for AI assistants, especially in private AI settings where models are expected to interpret users’ personal data (e.g., conversations, app usage) to understand their background, preferences, and social context. However, due to privacy concerns, existing academic research…

Cited by 23SourcePDFScholar
2025

Poison as Cure: Visual Noise for Mitigating Object Hallucinations in LVMs

NeurIPS 2025poster

Large vision-language models (LVMs) extend large language models (LLMs) with visual perception capabilities, enabling them to process and interpret visual information. A major challenge compromising their reliability is object hallucination that LVMs may generate plausible but factually inaccurate i…

Cited by 0SourceScholar
2025

Segment then Splat: Unified 3D Open-Vocabulary Segmentation via Gaussian Splatting

NeurIPS 2025poster

Open-vocabulary querying in 3D space is crucial for enabling more intelligent perception in applications such as robotics, autonomous systems, and augmented reality. However, most existing methods rely on 2D pixel-level parsing, leading to multi-view inconsistencies and poor 3D object retrieval. Mor…

Cited by 0SourceScholar
2025

Text2Data: Low-Resource Data Generation with Textual Control

AAAI 2025technical

Natural language serves as a common and straightforward control signal for humans to interact seamlessly with machines. Recognizing the importance of this interface, the machine learning community is investing considerable effort in generating data that is semantically coherent with textual instruct…

2025

xLAM: A Family of Large Action Models to Empower AI Agent Systems

NAACL 2025long

Autonomous agents powered by large language models (LLMs) have attracted significant research interest. However, the open-source community faces many challenges in developing specialized models for agent tasks, driven by the scarcity of high-quality agent datasets and the absence of standard protoco…

2024

A Simple Low-bit Quantization Framework for Video Snapshot Compressive Imaging

ECCV 2024oral

"Video Snapshot Compressive Imaging (SCI) aims to use a low-speed 2D camera to capture high-speed scene as snapshot compressed measurements, followed by a reconstruction algorithm to reconstruct the high-speed video frames. State-of-the-art (SOTA) deep learning-based algorithms have achieved impress…

2024

APIGen: Automated PIpeline for Generating Verifiable and Diverse Function-Calling Datasets

NeurIPS 2024poster

The advancement of function-calling agent models requires diverse, reliable, and high-quality datasets. This paper presents APIGen, an automated data generation pipeline designed to synthesize high-quality datasets for function-calling applications. We leverage APIGen and collect 3,673 executable AP…

2024

Communication-Oriented Automatic Assessment System for Accented Spoken Chinese in Read-Aloud Tasks

ICASSP 2024accepted

The development of speech signal processing and deep learning has brought in many intelligent language learning tools. However, non-native Chinese learners (second-language or L2 learners) are often discouraged by language assessment applications on the market because of their accent. By contrast to…

Cited by 0SourceScholar
2024

Don't Judge by the Look: Towards Motion Coherent Video Representation

ICLR 2024poster

Current training pipelines in object recognition neglect Hue Jittering when doing data augmentation as it not only brings appearance changes that are detrimental to classification, but also the implementation is inefficient in practice. In this study, we investigate the effect of hue variance in the…

2024

HIVE: Harnessing Human Feedback for Instructional Visual Editing

CVPR 2024poster

Incorporating human feedback has been shown to be crucial to align text generated by large language models to human preferences. We hypothesize that state-of-the-art instructional image editing models where outputs are generated based on an input image and an editing instruction could similarly bene…

2024

How Do Transformers Learn In-Context Beyond Simple Functions? A Case Study on Learning with Representations

ICLR 2024poster

While large language models based on the transformer architecture have demonstrated remarkable in-context learning (ICL) capabilities, understandings of such capabilities are still in an early stage, where existing theory and mechanistic understanding focus mostly on simple scenarios such as learnin…

Cited by 61SourcePDFScholar
2024

Retroformer: Retrospective Large Language Agents with Policy Gradient Optimization

ICLR 2024spotlight

Recent months have seen the emergence of a powerful new trend in which large language models (LLMs) are augmented to become autonomous language agents capable of performing objective oriented multi-step tasks on their own, rather than merely responding to queries from human users. Most existing lang…

2024

Sample-Efficient Learning of POMDPs with Multiple Observations In Hindsight

ICLR 2024poster

This paper studies the sample-efficiency of learning in Partially Observable Markov Decision Processes (POMDPs), a challenging problem in reinforcement learning that is known to be exponentially hard in the worst-case. Motivated by real-world settings such as loading in game playing, we propose an e…

Cited by 8SourcePDFScholar
2024

Slicing Vision Transformer for Flexible Inference

NeurIPS 2024poster

Vision Transformers (ViT) is known for its scalability. In this work, we target to scale down a ViT to fit in an environment with dynamic-changing resource constraints. We observe that smaller ViTs are intrinsically the sub-networks of a larger ViT with different widths. Thus, we propose a general f…

2024

Sparse Enhanced Network: An Adversarial Generation Method for Robust Augmentation in Sequential Recommendation

AAAI 2024technical

Sequential Recommendation plays a significant role in daily recommendation systems, such as e-commerce platforms like Amazon and Taobao. However, even with the advent of large models, these platforms often face sparse issues in the historical browsing records of individual users due to new users joi…

2023

CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesis

ICLR 2023top-25%

Program synthesis strives to generate a computer program as a solution to a given problem specification, expressed with input-output examples or natural language descriptions. The prevalence of large language models advances the state-of-the-art for program synthesis, though limited training resourc…

2023

Fantastic Rewards and How to Tame Them: A Case Study on Reward Learning for Task-oriented Dialogue Systems

ICLR 2023poster

When learning task-oriented dialogue (ToD) agents, reinforcement learning (RL) techniques can naturally be utilized to train dialogue strategies to achieve user-specific goals. Prior works mainly focus on adopting advanced RL techniques to train the ToD agents, while the design of the reward functio…

2023

Hierarchical Network with Decoupled Knowledge Distillation for Speech Emotion Recognition

ICASSP 2023accepted

The goal of Speech Emotion Recognition (SER) is to enable computers to recognize the emotion category of a given utterance in the same way that humans do. The accuracy of SER is strongly dependent on the validity of the utterance-level representation obtained by the model. Nevertheless, the "dark kn…

Cited by 0SourceScholar
2023

Improved Online Conformal Prediction via Strongly Adaptive Online Learning

ICML 2023poster

We study the problem of uncertainty quantification via prediction sets, in an online setting where the data distribution may vary arbitrarily over time. Recent work develops *online conformal prediction* techniques that leverage regret minimization algorithms from the online learning literature to l…

2023

Iterative Soft Shrinkage Learning for Efficient Image Super-Resolution

ICCV 2023poster

Image super-resolution (SR) has witnessed extensive neural network designs from CNN to transformer architectures. However, prevailing SR models suffer from prohibitive memory footprint and intensive computations, which limits further deployment on edge devices. This work investigates the potential o…

Cited by 5PDFcodeScholar
2023

PTC-Net: Point-Wise Transformer With Sparse Convolution Network for Place Recognition

RA-L 2023

In the point-cloud-based place recognition area, the existing hybrid architectures combining both convolutional networks and transformers have shown promising performance. They mainly apply the voxel-wise transformer after the sparse convolution (SPConv). However, they can induce information loss by

Cited by 22SourcecodeScholar
2023

Real-Time Neural Light Field on Mobile Devices

CVPR 2023poster

Recent efforts in Neural Rendering Fields (NeRF) have shown impressive results on novel view synthesis by utilizing implicit neural representation to represent 3D scenes. Due to the process of volumetric rendering, the inference speed for NeRF is extremely slow, limiting the application scenarios of…

2023

Recovering a Molecule's 3D Dynamics from Liquid-phase Electron Microscopy Movies

ICCV 2023poster

The dynamics of biomolecules are crucial for our understanding of their functioning in living systems. However, current 3D imaging techniques, such as cryogenic electron microscopy (cryo-EM), require freezing the sample, which limits the observation of their conformational changes in real time. The…

Cited by 5PDFScholar
2023

SnapFusion: Text-to-Image Diffusion Model on Mobile Devices within Two Seconds

NeurIPS 2023poster

Text-to-image diffusion models can create stunning images from natural language descriptions that rival the work of professional artists and photographers. However, these models are large, with complex network architectures and tens of denoising iterations, making them computationally expensive and…

Cited by 174SourcePDFScholar
2023

Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selection

NeurIPS 2023oral

Neural sequence models based on the transformer architecture have demonstrated remarkable \emph{in-context learning} (ICL) abilities, where they can perform new tasks when prompted with training and test examples, without any parameter update to the model. This work first provides a comprehensive st…

2023

UniControl: A Unified Diffusion Model for Controllable Visual Generation In the Wild

NeurIPS 2023poster

Achieving machine autonomy and human control often represent divergent objectives in the design of interactive AI systems. Visual generative foundation models such as Stable Diffusion show promise in navigating these goals, especially when prompted with arbitrary languages. However, they often fall…

2022

Efficient and Differentiable Conformal Prediction with General Function Classes

ICLR 2022poster

Quantifying the data uncertainty in learning tasks is often done by learning a prediction interval or prediction set of the label given the input. Two commonly desired properties for learned prediction sets are \emph{valid coverage} and \emph{good efficiency} (such as low length or low cardinality).…

2022

Ensemble of Averages: Improving Model Selection and Boosting Performance in Domain Generalization

NeurIPS 2022accept

In Domain Generalization (DG) settings, models trained independently on a given set of training domains have notoriously chaotic performance on distribution shifted test domains, and stochasticity in optimization (e.g. seed) plays a big role. This makes deep learning models unreliable in real world…

2022

Learning Efficient Image Super-Resolution Networks via Structure-Regularized Pruning

ICLR 2022poster

Several image super-resolution (SR) networks have been proposed of late for efficient SR, achieving promising results. However, they are still not lightweight enough and neglect to be extended to larger networks. At the same time, model compression techniques, like neural architecture search and kno…

Cited by 69SourcePDFScholar
2022

Local calibration: metrics and recalibration

UAI 2022poster

Probabilistic classifiers output confidence scores along with their predictions, and these confidence scores should be calibrated, i.e., they should reflect the reliability of the prediction. Confidence scores that minimize standard metrics such as the expected calibration error (ECE) accurately mea…

Cited by 23SourcePDFScholar
2022

Policy Optimization for Markov Games: Unified Framework and Faster Convergence

NeurIPS 2022accept

This paper studies policy optimization algorithms for multi-agent reinforcement learning. We begin by proposing an algorithm framework for two-player zero-sum Markov Games in the full-information setting, where each iteration consists of a policy update step at each state using a certain matrix game…

Cited by 33SourcePDFScholar
2022

Privacy-Preserving Face Recognition with Learnable Privacy Budgets in Frequency Domain

ECCV 2022poster

"Face recognition technology has been used in many fields due to its high recognition accuracy, including the face unlocking of mobile devices, community access control systems, and city surveillance. As the current high accuracy is guaranteed by very deep network structures, facial images often nee…

2022

R2L: Distilling Neural Radiance Field to Neural Light Field for Efficient Novel View Synthesis

ECCV 2022poster

"Recent research explosion on Neural Radiance Field (NeRF) shows the encouraging potential to represent complex scenes with neural networks. One major drawback of NeRF is its prohibitive inference time: Rendering a single pixel requires querying the NeRF network hundreds of times. To resolve it, exi…

2022

Recent Advances on Neural Network Pruning at Initialization

IJCAI 2022poster

Neural network pruning typically removes connections or neurons from a pretrained converged model; while a new pruning paradigm, pruning at initialization (PaI), attempts to prune a randomly initialized network. This paper offers the first survey concentrated on this emerging pruning fashion. We fir…

2022

What Makes a "Good" Data Augmentation in Knowledge Distillation - A Statistical Perspective

NeurIPS 2022accept

Knowledge distillation (KD) is a general neural network training approach that uses a teacher model to guide the student model. Existing works mainly study KD from the network output side (e.g., trying to design a better KD loss function), while few have attempted to understand it from the input sid…

2021

Aligned Structured Sparsity Learning for Efficient Image Super-Resolution

NeurIPS 2021spotlight

Lightweight image super-resolution (SR) networks have obtained promising results with moderate model size. Many SR methods have focused on designing lightweight architectures, which neglect to further reduce the redundancy of network parameters. On the other hand, model compression techniques, like…

2021

Catastrophic Fisher Explosion: Early Phase Fisher Matrix Impacts Generalization

ICML 2021spotlight

The early phase of training a deep neural network has a dramatic effect on the local curvature of the loss function. For instance, using a small learning rate does not guarantee stable optimization because the optimization trajectory has a tendency to steer towards regions of the loss surface with i…

Cited by 80SourcePDFScholar
2021

Context Reasoning Attention Network for Image Super-Resolution

ICCV 2021poster

Deep convolutional neural networks (CNNs) are achieving great successes for image super-resolution (SR), where global context is crucial for accurate restoration. However, the basic convolutional layer in CNNs is designed to extract local patterns, lacking the ability to model global context. Many e…

Cited by 89PDFScholar
2021

Don’t Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classification

ICML 2021spotlight

Modern machine learning models with high accuracy are often miscalibrated—the predicted top probability does not reflect the actual accuracy, and tends to be \emph{over-confident}. It is commonly believed that such over-confidence is mainly due to \emph{over-parametrization}, in particular when the…

Cited by 64SourcePDFScholar
2021

Evaluating State-of-the-Art Classification Models Against Bayes Optimality

NeurIPS 2021poster

Evaluating the inherent difficulty of a given data-driven classification problem is important for establishing absolute benchmarks and evaluating progress in the field. To this end, a natural quantity to consider is the \emph{Bayes error}, which measures the optimal classification error theoreticall…

Cited by 12SourcePDFScholar
2021

How Important is the Train-Validation Split in Meta-Learning?

ICML 2021spotlight

Meta-learning aims to perform fast adaptation on a new task through learning a “prior” from multiple existing tasks. A common practice in meta-learning is to perform a train-validation split (\emph{train-val method}) where the prior adapts to the task on one split of the data, and the resulting pred…

Cited by 92SourcePDFScholar
2021

Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement Learning

NeurIPS 2021poster

Recent theoretical work studies sample-efficient reinforcement learning (RL) extensively in two settings: learning interactively in the environment (online RL), or learning from an offline dataset (offline RL). However, existing algorithms and theories for learning near-optimal policies in these two…

Cited by 196SourcePDFScholar
2021

Sample-Efficient Learning of Stackelberg Equilibria in General-Sum Games

NeurIPS 2021poster

Real world applications such as economics and policy making often involve solving multi-agent games with two unique features: (1) The agents are inherently *asymmetric* and partitioned into leaders and followers; (2) The agents have different reward functions, thus the game is *general-sum*. The maj…

Cited by 83SourcePDFScholar
2021

Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation

NeurIPS 2021poster

Graph Similarity Computation (GSC) is essential to wide-ranging graph applications such as retrieval, plagiarism/anomaly detection, etc. The exact computation of graph similarity, e.g., Graph Edit Distance (GED), is an NP-hard problem that cannot be exactly solved within an adequate time given large…

2021

Understanding the Under-Coverage Bias in Uncertainty Estimation

NeurIPS 2021spotlight

Estimating the data uncertainty in regression tasks is often done by learning a quantile function or a prediction interval of the true label conditioned on the input. It is frequently observed that quantile regression---a vanilla algorithm for learning quantiles with asymptotic guarantees---tends to…

Cited by 18SourcePDFScholar
2021

Unsupervised Paraphrasing with Pretrained Language Models

EMNLP 2021main

Paraphrase generation has benefited extensively from recent progress in the designing of training objectives and model architectures. However, previous explorations have largely focused on supervised methods, which require a large amount of labeled data that is costly to collect. To address this dra…

2020

Assessing Local Generalization Capability in Deep Models

AISTATS 2020poster

While it has not yet been proven, empirical evidence suggests that model generalization is related to local properties of the optima, which can be described via the Hessian. We connect model generalization with the local property of a solution under the PAC-Bayes paradigm. In particular, we prove th…

2020

Collaborative Distillation for Ultra-Resolution Universal Style Transfer

CVPR 2020poster

Universal style transfer methods typically leverage rich representations from deep Convolutional Neural Network (CNN) models (e.g., VGG-19) pre-trained on large collections of images. Despite the effectiveness, its application is heavily constrained by the large model size to handle ultra-resolution…

Cited by 134PDFcodeScholar
2020

Towards Understanding Hierarchical Learning: Benefits of Neural Representations

NeurIPS 2020poster

Deep neural networks can empirically perform efficient hierarchical learning, in which the layers learn useful representations of the data. However, how they make use of the intermediate representations are not explained by recent theories that relate them to ``shallow learners'' such as kernels. In…

Cited by 63SourcePDFScholar
2019

Miss Detection vs. False Alarm: Adversarial Learning for Small Object Segmentation in Infrared Images

ICCV 2019poster

A key challenge of infrared small object segmentation (ISOS) is to balance miss detection (MD) and false alarm (FA). This usually needs "opposite" strategies to suppress the two terms, and has not been well resolved in the literature. In this paper, we propose a deep adversarial learning framework t…

Cited by 412PDFScholar
2019

On the Generalization Gap in Reparameterizable Reinforcement Learning

ICML 2019oral

Understanding generalization in reinforcement learning (RL) is a significant challenge, as many common assumptions of traditional supervised learning theory do not apply. We focus on the special class of reparameterizable RL problems, where the trajectory distribution can be decomposed using the rep…

Cited by 48SourcePDFScholar