← Search

TAO YANG

72 accepted papers

2026

AgentMath: Empowering Mathematical Reasoning for Large Language Models via Tool-Augmented Agent

ICLR 2026poster

Large Reasoning Models (LRMs) like o3 and DeepSeek-R1 have achieved remarkable progress in natural language reasoning with long chain-of-thought. However, they remain computationally inefficient and struggle with accuracy when solving problems requiring complex mathematical operations. In this work,…

Cited by 0SourceScholar
2026

HiNCoT: Hierarchical Nonlinear Continuous Transform-based Tensor Representation for Multi-Dimensional Data Recovery

AAAI 2026technical

Recently, continuous transform-based tensor representation has emerged as a promising tool for multi-dimensional data recovery. However, the existing continuous transforms are essentially single-layer linear mappings, which limits their ability to capture the complex relationships inherent in multi

Cited by 0SourcePDFScholar
2026

Inconsistency Biases in Dynamic Data Pruning

ICLR 2026poster

Dynamic data pruning accelerates training by focusing on informative samples. However, comparing importance scores across different model states introduces inconsistency (score context drift), and variable selection rates bias gradient dynamics over time (temporal gradient bias). We introduce RePB (…

Cited by 0SourcecodeScholar
2026

KnapSpec: Self-Speculative Decoding via Adaptive Layer Selection as a Knapsack Problem

ICML 2026poster

Self-speculative decoding (SSD) accelerates LLM inference by skipping layers to create an efficient draft model, yet existing methods often rely on static heuristics that ignore the dynamic computational overhead of attention in long-context scenarios. We propose KnapSpec, a training-free framework …

Cited by 0SourceScholar
2026

LeapAlign: Post-training Flow Matching Models at Any Generation Step by Building Two-Step Trajectories

CVPR 2026

This paper focuses on the alignment of flow-matching models with human preference. A promising way is fine-tuning by directly backpropagating reward signals through the differentiable generation process of flow matching. However, backpropagating through long trajectories results in prohibitive memor

Cited by 0SourcecodeScholar
2026

Learning More with Less: A Dynamic Dual-Level Down-Sampling Framework for Efficient Policy Optimization

ICLR 2026poster

Critic-free methods like GRPO reduce memory demands by estimating advantages from multiple rollouts but tend to converge slowly, as critical learning signals are diluted by an abundance of uninformative samples and tokens. To tackle this challenge, we propose the **Dynamic Dual-Level Down-Sampling (…

Cited by 0SourceScholar
2026

LighterBEV: LiDAR Global Localization Meets Online Learning

ICRA 2026poster

LiDAR-based global localization provides accurate robot pose estimates against a prior map. Existing deep-learning methods, however, demand heavy computation and long training or inference times and degrade sharply when faced with domain shifts. This letter presents LighterBEV, a lightweight, fast, …

2026

Many-for-Many: Unify the Training of Multiple Video and Image Generation and Manipulation Tasks

ICLR 2026poster

Diffusion models have shown impressive performance in many visual generation and manipulation tasks. Many existing methods focus on training a model for a specific task, especially, text-to-video (T2V) generation, while many other works focus on finetuning the pretrained T2V model for image-to-video…

Cited by 0SourcecodeScholar
2026

OCLPlace: Online Continual Learning on LiDAR Streams for Place Recognition

ICRA 2026poster

LiDAR place recognition is a critical component of LiDAR-based localization pipelines, tasked with identifying previously visited places across diverse environments and temporal conditions. A growing body of deep learning–based approaches has recently tackled this problem. However, their performance…

Cited by 0codeScholar
2026

Self-Forcing++: Towards Minute-Scale High-Quality Video Generation

ICLR 2026poster

Diffusion models have revolutionized image and video generation, achieving unprecedented visual quality. However, their reliance on transformer architectures incurs prohibitively high computational costs, particularly when extending generation to long videos. Recent work has explored autoregressive…

Cited by 0SourcecodeScholar
2025

AF-UMC: An Alignment-Free Fusion Framework for Unaligned Multi-View Clustering

NeurIPS 2025poster

The Unaligned Multi-view Clustering (UMC) aims to learn a discriminative cluster structure from unaligned multi-view data, where the features of samples are not completely aligned across multiple views. Most existing methods usually prioritize employing various alignment strategies to align sample r…

Cited by 0SourceScholar
2025

CaliGCL: Calibrated Graph Contrastive Learning via Partitioned Similarity and Consistency Discrimination

NeurIPS 2025poster

Graph contrastive learning (GCL) aims to learn self-supervised representations by distinguishing positive and negative sample pairs generated from multiple augmented graph views. Despite showing promising performance, GCL still suffers from two critical biases: (1) ***Similarity estimation bias*** a…

Cited by 0SourceScholar
2025

Discriminative Policy Optimization for Token-Level Reward Models

ICML 2025poster

Process reward models (PRMs) provide more nuanced supervision compared to outcome reward models (ORMs) for optimizing policy models, positioning them as a promising approach to enhancing the capabilities of LLMs in complex reasoning tasks. Recent efforts have advanced PRMs from step-level to token-l…

2025

Enabling In-Flight Metamorphosis in Multirotors with a Center-Driven Scissor Extendable Airframe for Adaptive Navigation

ICRA 2025

To address complex mission tasks, multirotors benefit from in-flight reconfiguration that enhances their morphological adaptability. This paper presents the Center-Driven Scissor Extendable Airframe (CDSEA), a novel one-degree-of-freedom (DOF) morphing airframe designed to replace traditional fixed-

Cited by 0SourceScholar
2025

Improving Large Language Models Function Calling and Interpretability via Guided-Structured Templates

EMNLP 2025

Large language models (LLMs) have demonstrated strong reasoning and tool-use capabilities, yet they often fail in real-world tool-interactions due to incorrect parameterization, poor tool selection, or misinterpretation of user intent. These issues often stem from an incomplete understanding of user

2025

MPS-Prover: Advancing Stepwise Theorem Proving by Multi-Perspective Search and Data Curation

NeurIPS 2025poster

Automated Theorem Proving (ATP) in formal languages remains a formidable challenge in AI, demanding rigorous logical deduction and navigating vast search spaces. While large language models (LLMs) have shown promising performance, existing stepwise provers often suffer from biased search guidance, l…

Cited by 0SourceScholar
2025

Multi-target Association and Localization with Distributed Drone Following: A Factor Graph Approach

IROS 2025

Vision-based multi-drone multi-object tracking technology enables autonomous target situational awareness for unmanned aerial systems. Distributed observer drones dynamically estimate the spatio-temporal states of multiple targets through collaborative sensor fusion, enabling simultaneous localizati

Cited by 0SourcecodeScholar
2025

PaceLLM: Brain-Inspired Large Language Models for Long-Context Understanding

NeurIPS 2025poster

While Large Language Models (LLMs) demonstrate strong performance across domains, their long-context capabilities are limited by transient neural activations causing information decay and unstructured feed-forward network (FFN) weights leading to semantic fragmentation. Inspired by the brain’s worki…

Cited by 0SourceScholar
2025

ProSparse: Introducing and Enhancing Intrinsic Activation Sparsity within Large Language Models

COLING 2025main

Activation sparsity refers to the existence of considerable weakly-contributed elements among activation outputs, serving as a promising paradigm for accelerating model inference. Nevertheless, most large language models (LLMs) adopt activation functions without intrinsic activation sparsity (e.g.,…

2025

Prompt-augmented Feature with Cross-domain Contrastive Learning for Efficient Multi-domain Sentiment Analysis

ICASSP 2025accepted

Pre-trained language models (PrLMs) demonstrate impressive performance on the sentiment analysis task. However, the large number of trainable parameters brings about heavy computational costs, which become more serious in multi-domain scenarios. In this paper, we propose to extract multi-layer featu…

Cited by 0SourceScholar
2024

Breaking through the learning plateaus of in-context learning in Transformer

ICML 2024poster

In-context learning, i.e., learning from context examples, is an impressive ability of Transformer. Training Transformers to possess this in-context learning skill is computationally intensive due to the occurrence of *learning plateaus*, which are periods within the training process where there is…

Cited by 1SourcePDFScholar
2024

Closed-Loop Unsupervised Representation Disentanglement with $\\beta$-VAE Distillation and Diffusion Probabilistic Feedback

ECCV 2024poster

"Representation disentanglement may help AI fundamentally understand the real world and thus benefit both discrimination and generation tasks. It currently has at least three unresolved core issues: (i) heavy reliance on label annotation and synthetic data — causing poor generalization on natural sc…

Cited by 7SourcePDFScholar
2024

Diffusion Model with Cross Attention as an Inductive Bias for Disentanglement

NeurIPS 2024spotlight

Disentangled representation learning strives to extract the intrinsic factors within the observed data. Factoring these representations in an unsupervised manner is notably challenging and usually requires tailored loss functions or specific structural designs. In this paper, we introduce a new pers…

Cited by 6SourcePDFScholar
2024

MicroCinema: A Divide-and-Conquer Approach for Text-to-Video Generation

CVPR 2024highlight

We present MicroCinema a straightforward yet effective framework for high-quality and coherent text-to-video generation. Unlike existing approaches that align text prompts with video directly MicroCinema introduces a Divide-and-Conquer strategy which divides the text-to-video into a two-stage proces…

Cited by 15SourcePDFScholar
2024

Preventing Catastrophic Forgetting in Continuous Online Learning for Autonomous Driving

IROS 2024poster

Autonomous vehicles require online learning capabilities to enable long-term, unattended operation. However, long-term online learning is accompanied by the problem of forgetting previously learned knowledge. This paper introduces an online learning framework that includes a catastrophic forgetting…

Cited by 3SourcecodeScholar
2024

Private Language Models via Truncated Laplacian Mechanism

EMNLP 2024main

Recently it has been shown that deep learning models for NLP tasks are prone to attacks that can even reconstruct the verbatim training texts. To prevent privacy leakage, researchers have investigated word-level perturbations, relying on the formal guarantees of differential privacy (DP) in the embe…

Cited by 1SourcePDFScholar
2024

RMSC-VIO: Robust Multi-Stereoscopic Visual-Inertial Odometry for Local Visually Challenging Scenarios

RA-L 2024

We present a Multi-Stereoscopic Visual-Inertial Odometry (VIO) system capable of integrating an arbitrary number of stereo cameras, exhibiting excellent robustness in the face of visually challenging scenarios. During system initialization, we introduce multi-view keyframes for simultaneous processi

Cited by 6SourceScholar
2024

Robust and Scalable Model Editing for Large Language Models

COLING 2024main

Large language models (LLMs) can make predictions using *parametric knowledge* – knowledge encoded in the model weights – or *contextual knowledge* – knowledge presented in the context. In many scenarios, a desirable behavior is that LLMs give precedence to contextual knowledge when it conflicts wit…

2024

SeeSR: Towards Semantics-Aware Real-World Image Super-Resolution

CVPR 2024poster

Owe to the powerful generative priors the pre-trained text-to-image (T2I) diffusion models have become increasingly popular in solving the real-world image super-resolution problem. However as a consequence of the heavy quality degradation of input low-resolution (LR) images the destruction of local…

2024

Threshold-driven Pruning with Segmented Maximum Term Weights for Approximate Cluster-based Sparse Retrieval

EMNLP 2024main

This paper revisits dynamic pruning through rank score thresholding in cluster-based sparse retrieval to skip the index partially at cluster and document levels during inference. It proposes a two-parameter pruning control scheme called ASC with a probabilistic guarantee on rank-safeness competitive…

2023

Clustering-Aware Negative Sampling for Unsupervised Sentence Representation

ACL 2023findings

Contrastive learning has been widely studied in sentence representation learning. However, earlier works mainly focus on the construction of positive examples, while in-batch samples are often simply treated as negative examples. This approach overlooks the importance of selecting appropriate negati…

2023

DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

ICCV 2023poster

Image colorization is a challenging problem due to multi-modal uncertainty and high ill-posedness. Directly training a deep neural network usually leads to incorrect semantic colors and low color richness. While transformer-based methods can deliver better results, they often rely on manually design…

Cited by 64PDFcodeScholar
2023

DisDiff: Unsupervised Disentanglement of Diffusion Probabilistic Models

NeurIPS 2023poster

Targeting to understand the underlying explainable factors behind observations and modeling the conditional generation process on these factors, we connect disentangled representation learning to diffusion probabilistic models (DPMs) to take advantage of the remarkable modeling ability of DPMs. We p…

2023

Dual-Feedback Knowledge Retrieval for Task-Oriented Dialogue Systems

EMNLP 2023long main

Efficient knowledge retrieval plays a pivotal role in ensuring the success of end-to-end task-oriented dialogue systems by facilitating the selection of relevant information necessary to fulfill user requests. However, current approaches generally integrate knowledge retrieval and response generatio…

Cited by 0SourceScholar
2023

Explore-Instruct: Enhancing Domain-Specific Instruction Coverage through Active Exploration

EMNLP 2023long main

Instruction-tuning can be substantially optimized through enhanced diversity, resulting in models capable of handling a broader spectrum of tasks. However, existing data employed for such tuning often exhibit an inadequate coverage of individual domains, limiting the scope for nuanced comprehension…

Cited by 0SourcecodeScholar
2023

Orders Are Unwanted: Dynamic Deep Graph Convolutional Network for Personality Detection

AAAI 2023technical

Predicting personality traits based on online posts has emerged as an important task in many fields such as social network analysis. One of the challenges of this task is assembling information from various posts into an overall profile for each user. While many previous solutions simply concatenate…

2023

PsyCoT: Psychological Questionnaire as Powerful Chain-of-Thought for Personality Detection

EMNLP 2023long findings

Recent advances in large language models (LLMs), such as ChatGPT, have showcased remarkable zero-shot performance across various NLP tasks. However, the potential of LLMs in personality detection, which involves identifying an individual's personality from their written texts, remains largely unexpl…

Cited by 0SourcecodeScholar
2022

AD-DROP: Attribution-Driven Dropout for Robust Language Model Fine-Tuning

NeurIPS 2022accept

Fine-tuning large pre-trained language models on downstream tasks is apt to suffer from overfitting when limited training data is available. While dropout proves to be an effective antidote by randomly dropping a proportion of units, existing research has not examined its effect on the self-attentio…

2022

Actor-Critic Policy Optimization in a Large-Scale Imperfect-Information Game

ICLR 2022poster

The deep policy gradient method has demonstrated promising results in many large-scale games, where the agent learns purely from its own experience. Yet, policy gradient methods with self-play suffer convergence problems to a Nash Equilibrium (NE) in multi-agent situations. Counterfactual regret min…

Cited by 33SourcePDFScholar
2022

CRASpell: A Contextual Typo Robust Approach to Improve Chinese Spelling Correction

ACL 2022findings

Recently, Bert-based models have dominated the research of Chinese spelling correction (CSC). These methods have two limitations: (1) they have poor performance on multi-typo texts. In such texts, the context of each typo contains at least one misspelled character, which brings noise information. Su…

2022

Compact Token Representations with Contextual Quantization for Efficient Document Re-ranking

ACL 2022long

Transformer based re-ranking models can achieve high search relevance through context- aware soft matching of query tokens with document tokens. To alleviate runtime complexity of such inference, previous work has adopted a late interaction architecture with pre-computed contextual token representat…

2022

Contextual Debiasing for Visual Recognition With Causal Mechanisms

CVPR 2022poster

As a common problem in the visual world, contextual bias means the recognition may depend on the co-occurrence context rather than the objects themselves, which is even more severe in multi-label tasks due to multiple targets and the absence of location. Although some studies have focused on tacklin…

Cited by 46PDFcodeScholar
2022

Improving Chinese Grammatical Error Detection via Data augmentation by Conditional Error Generation

ACL 2022findings

Chinese Grammatical Error Detection(CGED) aims at detecting grammatical errors in Chinese texts. One of the main challenges for CGED is the lack of annotated data. To alleviate this problem, previous studies proposed various methods to automatically generate more training samples, which can be rough…

2022

Learning Disentangled Representation by Exploiting Pretrained Generative Models: A Contrastive Learning View

ICLR 2022poster

From the intuitive notion of disentanglement, the image variations corresponding to different generative factors should be distinct from each other, and the disentangled representation should reflect those variations with separate dimensions. To discover the generative factors and learn disentangled…

2022

Towards Building A Group-based Unsupervised Representation Disentanglement Framework

ICLR 2022poster

Disentangled representation learning is one of the major goals of deep learning, and is a key step for achieving explainable and generalizable models. The key idea of the state-of-the-art VAE-based unsupervised representation disentanglement methods is to minimize the total correlation of the joint…

2021

KLMo: Knowledge Graph Enhanced Pretrained Language Model with Fine-Grained Relationships

EMNLP 2021finding

Interactions between entities in knowledge graph (KG) provide rich knowledge for language representation learning. However, existing knowledge-enhanced pretrained language models (PLMs) only focus on entity information and ignore the fine-grained relationships between entities. In this work, we prop…

2021

Learning to Answer Psychological Questionnaire for Personality Detection

EMNLP 2021finding

Existing text-based personality detection research mostly relies on data-driven approaches to implicitly capture personality cues in online posts, lacking the guidance of psychological knowledge. Psychological questionnaire, which contains a series of dedicated questions highly related to personalit…

Cited by 22SourcePDFScholar
2021

Multiplex Graph Neural Network for Extractive Text Summarization

EMNLP 2021main

Extractive text summarization aims at extracting the most representative sentences from a given document as its summary. To extract a good summary from a long text document, sentence embedding plays an important role. Recent studies have leveraged graph neural networks to capture the inter-sententia…

Cited by 45SourcePDFScholar
2021

PLOME: Pre-training with Misspelled Knowledge for Chinese Spelling Correction

ACL 2021long

Chinese spelling correction (CSC) is a task to detect and correct spelling errors in texts. CSC is essentially a linguistic problem, thus the ability of language understanding is crucial to this task. In this paper, we propose a Pre-trained masked Language model with Misspelled knowledgE (PLOME) for…

2021

Psycholinguistic Tripartite Graph Network for Personality Detection

ACL 2021long

Most of the recent work on personality detection from online posts adopts multifarious deep neural networks to represent the posts and builds predictive models in a data-driven manner, without the exploitation of psycholinguistic knowledge that may unveil the connections between one’s language use a…

Cited by 42SourcePDFScholar
2021

Regret and Cumulative Constraint Violation Analysis for Online Convex Optimization with Long Term Constraints

ICML 2021oral

This paper considers online convex optimization with long term constraints, where constraints can be violated in intermediate rounds, but need to be satisfied in the long run. The cumulative constraint violation is used as the metric to measure constraint violations, which excludes the situation tha…

Cited by 52SourcePDFScholar
2021

Safety-Aware Motion Prediction With Unseen Vehicles for Autonomous Driving

ICCV 2021poster

Motion prediction of vehicles is critical but challenging due to the uncertainties in complex environments and the limited visibility caused by occlusions and limited sensor ranges. In this paper, we study a new task, safety-aware motion prediction with unseen vehicles for autonomous driving. Unlike…

Cited by 32PDFcodeScholar
2020

LaNoising: A Data-driven Approach for 903nm ToF LiDAR Performance Modeling under Fog

IROS 2020poster

As a critical sensor for high-level autonomous vehicles, LiDAR's limitations in adverse weather (e.g. rain, fog, snow, etc.) impede the deployment of self-driving cars in all weather conditions. In this paper, we model the performance of a popular 903nm ToF LiDAR under various fog conditions based o…

Cited by 31SourceScholar
2020

SNIAE-SSE Deformation Mechanism Enabled Scalable Multicopter: Design, Modeling and Flight Performance Validation

ICRA 2020poster

This paper focuses on designing, modeling and validating a novel scalable multicopter whose deformation mechanism, called SNIAE-SSE, relies on a combination of simple non-intersecting angulated elements (SNIAEs) and straight scissor-like elements (SSEs). The proposed SNIAE-SSE mechanism has the adva…

Cited by 14SourceScholar
2017

Efficient hybrid space-ground precoding techniques for multi-beam satellite systems

ICASSP 2017accepted

Multi-beam mobile satellite systems aim at providing broadband and high speed mobile services over a large area to achieve a high system throughput, where hybrid space-ground beamforming is one of the most promising candidates for ground-based beamforming techniques. It not only reduces the feeder l…

Cited by 0SourceScholar
2017

Robust multichannel TDOA estimation for speaker localization using the impulsive characteristics of speech spectrum

ICASSP 2017accepted

Time delay estimation (TDE) plays an important role in localizing and tracking radiating acoustic sources. Although many efforts have been devoted to this problem in the literature, the robustness of TDE with respect to noise and reverberation remains a great challenge for practical systems. In this…

Cited by 0SourceScholar
2016

On time delay estimation based on multichannel spatiotemporal sparse linear prediction

ICASSP 2016accepted

Noise and reverberation can significantly affect the performance of time delay estimation (TDE) in room acoustic environments. The multichannel cross-correlation coefficient (MCCC) algorithm, which extends the traditional cross-correlation method from two to multiple channels, can exploit the spatia…

Cited by 0SourceScholar
2016

Piecewise sparse signal recovery via piecewise orthogonal matching pursuit

ICASSP 2016accepted

In this paper, we consider the recovery of piecewise sparse signals from incomplete noisy measurements via a greedy algorithm. Here piecewise sparse means that the signal can be approximated in certain domain with known number of nonzero entries in each piece/segment. This paper makes a two-fold con…

Cited by 0SourceScholar