← Search

Yuan Cheng

39 accepted papers

2026

AgentXRay: White-Boxing Agentic Systems via Workflow Reconstruction

ICML 2026poster

Large Language Models have shown strong capabilities in complex problem solving, yet many agentic systems remain difficult to interpret and control due to opaque internal workflows. While some frameworks offer explicit architectures for collaboration, many deployed agentic systems operate as black b…

Cited by 0SourceScholar
2026

Beyond Instance-Level Self-Supervision in 3D Multi-Modal Medical Imaging

ICML 2026poster

Self-supervised pre-training methods in medical imaging typically treat each individual as an isolated instance, learning representations through augmentation-based objectives or masked reconstruction. They often do not adequately capitalize on a key characteristic of physiological features: anatomi…

Cited by 0SourceScholar
2026

Disco: Densely-overlapping Cell Instance Segmentation via Adjacency-aware Collaborative Coloring

ICLR 2026poster

Accurate cell instance segmentation is foundational for digital pathology analysis. Existing methods based on contour detection and distance mapping still face significant challenges in processing complex and dense cellular regions. Graph coloring-based methods provide a new paradigm for this task,…

Cited by 0SourcecodeScholar
2026

DistFlow: A Fully Distributed RL Framework for Scalable and Efficient LLM Post-Training

ICML 2026poster

Effectively scaling Reinforcement Learning (RL) is crucial for enhancing the reasoning and alignment of Large Language Models. The massive data and complex execution flows inherent in these tasks require a distributed architecture capable of efficient scaling. However, to simplify programming and de…

Cited by 0SourceScholar
2026

FLAG: Foundation model representation with Latent diffusion Alignment via Graph for spatial gene expression prediction

ICML 2026poster

Predicting spatial gene expression from routine H\&E makes high-resolution molecular profiling accessible at scale, especially for large retrospective cohorts. However, current models mostly treat gene expression as a series of pointwise tasks. While effective for numerical fitting, this approach ov…

Cited by 0SourceScholar
2026

Hear What You See: Video-to-Audio Generation with Diffusion Transformer and Semantic-Temporal Alignment-Ranked Direct Preference Optimization

CVPR 2026

Generating high-fidelity audio that is both semantically meaningful and temporally synchronized with silent videos remains a challenging problem in video-to-audio generation. Existing approaches often fail to capture fine-grained temporal correspondence between visual events and audio dynamics, lead

Cited by 0SourcecodeScholar
2026

Metis: Training LLMs with FP4 Quantization

ICLR 2026poster

This work identifies anisotropy in the singular value spectra of parameters, activations, and gradients as the fundamental barrier to low-bit training of large language models (LLMs). These spectra are dominated by a small fraction of large singular values, inducing wide numerical ranges that cause…

Cited by 0SourceScholar
2026

Multi-Head Attention as a Source of Catastrophic Forgetting in MoE Transformers

ICML 2026poster

Mixture-of-Experts (MoE) architectures are often considered a natural fit for continual learning because sparse routing should localize updates and reduce interference, yet MoE Transformers still forget substantially even with sparse, well-balanced expert utilization. We attribute this gap to a pre-…

Cited by 0SourceScholar
2026

PET2Rep: Towards Vision-Language Model-Drived Automated Radiology Report Generation for Positron Emission Tomography

AAAI 2026technical

Positron emission tomography (PET) is a cornerstone of modern oncologic and neurologic imaging, distinguished by its unique ability to illuminate dynamic metabolic processes that transcend the anatomical focus of traditional imaging technologies. Radiology reports are essential for clinical decision

Cited by 0SourcePDFScholar
2026

SAES-SVD: Self-Adaptive Suppression of Accumulated and Local Errors for SVD-based LLM Compression

ICLR 2026poster

The rapid growth in the parameter scale of large language models (LLMs) has created a high demand for efficient compression techniques. As a hardware-agnostic and highly compatible technique, low-rank compression has been widely adopted. However, existing methods typically compress each layer indep…

Cited by 0SourceScholar
2026

SD-MoE: Spectral Decomposition for Effective Expert Specialization

ICML 2026poster

Mixture-of-Experts (MoE) architectures scale Large Language Models via expert specialization induced by conditional computation. In practice, however, expert specialization often fails: some experts become functionally similar, while others functioning as de facto shared experts, limiting the effect…

Cited by 0SourceScholar
2026

Spectra: Rethinking Optimizers for LLMs Under Spectral Anisotropy

ICML 2026poster

Gradient signals in LLM training are highly anisotropic: recurrent linguistic structure concentrates energy into a small set of dominant spectral directions, while context-specific information resides in a long tail. We show that this spike–tail separation persists throughout training, with the spik…

Cited by 0SourceScholar
2026

Sprint or Delve: A Distribution-Aware Approach to Efficient Reasoning

IJCAI 2026

Reasoning chains in Large Language Models (LLMs) often exhibit heavy-tailed length distributions, yet existing efficiency methods rely on suboptimal linear penalties that suppress complex reasoning, limiting both accuracy and generalization. To address this, we first empirically observe that reasoni

Cited by 0Scholar
2026

Structure-based RNA Design by Step-wise Optimization of Latent Diffusion Model

AAAI 2026technical

RNA inverse folding, designing sequences to form specific 3D structures, is critical for therapeutics, gene regulation, and synthetic biology. Current methods, focused on sequence recovery, struggle to address structural objectives like secondary structure consistency (SS), minimum free energy (MFE)

Cited by 0SourcePDFScholar
2026

Tracing the Heart’s Pathways: ECG Representation Learning from a Cardiac Conduction Perspective

AAAI 2026technical

The multi-lead electrocardiogram (ECG) stands as a cornerstone of cardiac diagnosis. Recent strides in electrocardiogram self-supervised learning (eSSL) have brightened prospects for enhancing representation learning without relying on high-quality annotations. Yet earlier eSSL methods suffer a key

Cited by 0SourcePDFScholar
2026

UST-Hand: An Uncertainty-aware Spatiotemporal Point Cloud Interaction Network for 3D Self-supervised Hand Pose Estimation

CVPR 2026

Manually annotating accurate 3D hand poses is extremely time-consuming and labor-intensive. Existing self-supervised hand pose estimation methods leverage the discrepancy between input images and rendered outputs, or multi-view consistency constraints, as the driving force to optimize networks and p

Cited by 0SourceScholar
2025

ChromFound: Towards A Universal Foundation Model for Single-Cell Chromatin Accessibiltiy Data

NeurIPS 2025poster

The advent of single-cell Assay for Transposase-Accessible Chromatin using sequencing (scATAC-seq) offers an innovative perspective for deciphering regulatory mechanisms by assembling a vast repository of single-cell chromatin accessibility data. While foundation models have achieved significant suc…

Cited by 0SourcecodeScholar
2025

CogniDual Framework: Self-Training Large Language Models within a Dual-System Theoretical Framework for Improving Cognitive Tasks

ICASSP 2025accepted

Cognitive psychology investigates perception, attention, memory, language, problem-solving, decision-making, and reasoning. Kahneman’s dual-system theory elucidates the human decision-making process, distinguishing between the rapid, intuitive System 1 and the deliberative, rational System 2. Recent…

Cited by 0SourceScholar
2025

Efficient Network Automatic Relevance Determination

ICML 2025poster

We propose Network Automatic Relevance Determination (NARD), an extension of ARD for linearly probabilistic models, to simultaneously model sparse relationships between inputs $X \in \mathbb R^{d \times N}$ and outputs $Y \in \mathbb R^{m \times N}$, while capturing the correlation structure among t…

Cited by 0SourcePDFScholar
2025

Minimal Semantic Sufficiency Meets Unsupervised Domain Generalization

NeurIPS 2025poster

The generalization ability of deep learning has been extensively studied in supervised settings, yet it remains less explored in unsupervised scenarios. Recently, the Unsupervised Domain Generalization (UDG) task has been proposed to enhance the generalization of models trained with prevalent unsupe…

Cited by 0SourceScholar
2025

OSTQuant: Refining Large Language Model Quantization with Orthogonal and Scaling Transformations for Better Distribution Fitting

ICLR 2025poster

Post-training quantization (PTQ) has emerged as a widely adopted technique for compressing and accelerating Large Language Models (LLMs). The major challenge in LLM quantization is that uneven and heavy-tailed data distributions can expand the quantization range, thereby reducing bit precision for m…

2025

Pushing the Limits of BFP on Narrow Precision LLM Inference

AAAI 2025technical

The substantial computational and memory demands of Large Language Models (LLMs) hinder their deployment. Block Floating Point (BFP) has proven effective in accelerating linear operations, a cornerstone of LLM workloads. However, as sequence lengths grow, nonlinear operations, such as Attention, in…

Cited by 0SourcePDFScholar
2025

SegAnyPET: Universal Promptable Segmentation from Positron Emission Tomography Images

ICCV 2025poster

Positron Emission Tomography (PET) is a powerful molecular imaging tool that plays a crucial role in modern medical diagnostics by visualizing radio-tracer distribution to reveal physiological processes. Accurate organ segmentation from PET images is essential for comprehensive multi-systemic analys…

2025

Structure-aware Semantic Discrepancy and Consistency for 3D Medical Image Self-supervised Learning

ICCV 2025poster

3D medical image self-supervised learning (mSSL) holds great promise for medical analysis. Effectively supporting broader applications requires considering anatomical structure variations in location, scale, and morphology, which are crucial for capturing meaningful distinctions. However, previous m…

2025

Towards a Universal 3D Medical Multi-modality Generalization via Learning Personalized Invariant Representation

ICCV 2025poster

Variations in medical imaging modalities and individual anatomical differences pose challenges to cross-modality generalization in multi-modal tasks. Existing methods often concentrate exclusively on common anatomical patterns, thereby neglecting individual differences and consequently limiting thei…

2024

Provable Benefits of Multi-task RL under Non-Markovian Decision Making Processes

ICLR 2024poster

In multi-task reinforcement learning (RL) under Markov decision processes (MDPs), the presence of shared latent structures among multiple MDPs has been shown to yield significant benefits to the sample efficiency compared to single-task RL. In this paper, we investigate whether such a benefit can ex…

Cited by 1SourcePDFScholar
2024

Sample Complexity Characterization for Linear Contextual MDPs

AISTATS 2024poster

Contextual Markov decision processes (CMDPs) describe a class of reinforcement learning problems in which the transition kernels and reward functions can change over time with different MDPs indexed by a context variable. While CMDPs serve as an important framework to model many real-world applicati…

Cited by 1SourcePDFScholar
2024

ULMR: Unlearning Large Language Models via Negative Response and Model Parameter Average

EMNLP 2024industry

In recent years, large language models (LLMs) have attracted significant interest from the research community due to their broad applicability in many language-oriented tasks, and are now widely used in numerous areas of production and daily life. One source of the powerful capabilities of LLMs is t…

Cited by 1SourcePDFScholar
2023

Boundary-Aware Backward-Compatible Representation via Adversarial Learning in Image Retrieval

CVPR 2023poster

Image retrieval plays an important role in the Internet world. Usually, the core parts of mainstream visual retrieval systems include an online service of the embedding model and a large-scale vector database. For traditional model upgrades, the old model will not be replaced by the new one until th…

2023

DC-Former: Diverse and Compact Transformer for Person Re-identification

AAAI 2023technical

In person re-identification (ReID) task, it is still challenging to learn discriminative representation by deep learning, due to limited data. Generally speaking, the model will get better performance when increasing the amount of data. The addition of similar classes strengthens the ability of the…

2023

Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPs

ICLR 2023poster

In reward-free reinforcement learning (RL), an agent explores the environment first without any reward information, in order to achieve certain learning goals afterwards for any given reward. In this paper we focus on reward-free RL under low-rank MDP models, in which both the representation and lin…

Cited by 11SourcePDFScholar
2022

A Large-Scale Comprehensive Dataset and Copy-Overlap Aware Evaluation Protocol for Segment-Level Video Copy Detection

CVPR 2022poster

In this paper, we introduce VCSL (Video Copy Segment Localization), a new comprehensive segment-level annotated video copy dataset. Compared with existing copy detection datasets restricted by either video-level annotation or small-scale, VCSL not only has two orders of magnitude more segment-level…

Cited by 18PDFcodeScholar
2022

Deeply Tensor Compressed Transformers for End-to-End Object Detection

AAAI 2022technical

DEtection TRansformer (DETR) is a recently proposed method that streamlines the detection pipeline and achieves competitive results against two-stage detectors such as Faster-RCNN. The DETR models get rid of complex anchor generation and post-processing procedures thereby making the detection pipeli…

Cited by 15SourcePDFScholar
2022

Provable Benefit of Multitask Representation Learning in Reinforcement Learning

NeurIPS 2022accept

As representation learning becomes a powerful technique to reduce sample complexity in reinforcement learning (RL) in practice, theoretical understanding of its advantage is still limited. In this paper, we theoretically characterize the benefit of representation learning under the low-rank Markov d…

Cited by 28SourcePDFScholar
2022

Stacked Hybrid-Attention and Group Collaborative Learning for Unbiased Scene Graph Generation

CVPR 2022poster

Scene Graph Generation, which generally follows a regular encoder-decoder pipeline, aims to first encode the visual contents within the given image and then parse them into a compact summary graph. Existing SGG approaches generally not only neglect the insufficient modality fusion between vision and…

Cited by 127PDFcodeScholar
2021

Discrimination-Aware Mechanism for Fine-Grained Representation Learning

CVPR 2021poster

Recently, with the emergence of retrieval requirements for certain individual in the same superclass, e.g., birds, persons, cars, fine-grained recognition task has attracted a significant amount of attention from academia and industry. In fine-grained recognition scenario, the inter-class difference…

Cited by 24PDFScholar