← Search

Wei Cheng

70 accepted papers

2026

Escaping Whack-a-Mole: Code Documentation Optimization via Dependency-Guided Bi-level Search

ICML 2026poster

As large language models increasingly serve as autonomous coding agents, code documentation must be optimized for agent comprehension rather than human readability. We frame agent-oriented documentation generation as a black-box optimization problem over the documentation space, where quality is mea…

Cited by 0SourceScholar
2026

Improving Code Translation with Syntax-Guided and Semantic-aware Preference Optimization

IJCAI 2026

LLMs have shown immense potential for code translation, yet they often struggle to ensure both syntactic correctness and semantic consistency. While preference-based learning offers a promising alignment strategy, it is hindered by unreliable semantic rewards derived from sparse test cases or restri

Cited by 0Scholar
2026

OmniLottie: Generating Vector Animations via Parameterized Lottie Tokens

CVPR 2026

OmniLottie is a versatile framework that generates high-quality vector animations from multi-modal instructions, including interleaved texts, images, and videos. To fully parameterize vector animations for flexible motion and visual content control, we seek help from the Lottie representation, which

Cited by 0SourcecodeScholar
2026

ReasonEdit: Towards Reasoning-Enhanced Image Editing Models

CVPR 2026

Recent advances in image editing models have shown remarkable progress. A common architectural design couples a multimodal large language model (MLLM) encoder with a diffusion decoder, as seen in systems such as Step1X-Edit and Qwen-Image-Edit, where the MLLM encodes both the reference image and the

Cited by 0SourcecodeScholar
2026

RegionE: Adaptive Region-Aware Generation for Efficient Image Editing

ICLR 2026poster

Recently, instruction-based image editing (IIE) has received widespread attention. In practice, IIE often modifies only specific regions of an image, while the remaining areas largely remain unchanged. Although these two types of regions differ significantly in generation difficulty and computationa…

Cited by 0SourceScholar
2026

SkillGen: Learning Domain Skills for In-Context Sequential Decision Making

AAAI 2026technical

Large language models (LLMs) are increasingly applied to sequential decision-making through in-context learning (ICL), yet their effectiveness is highly sensitive to prompt quality. Effective prompts should meet three principles: focus on decision-critical information, provide step-level granularity

Cited by 0SourcePDFScholar
2026

Sparse-vDiT: Unleashing the Power of Sparse Attention to Accelerate Video Diffusion Transformers

AAAI 2026technical

While Diffusion Transformers (DiTs) have achieved breakthroughs in video generation, this long sequence generation task remains constrained by the quadratic complexity of attention mechanisms, resulting in significant inference latency. Through detailed analysis of attention maps in Video Diffusion

Cited by 0SourcePDFScholar
2026

ViStoryBench: Comprehensive Benchmark Suite for Story Visualization

CVPR 2026

Story visualization aims to generate coherent image sequences that faithfully represent a narrative and match given character references. Despite progress in generative models, existing benchmarks remain narrow in scope, often limited to short prompts, lacking character references, or single-image c

Cited by 0SourcecodeScholar
2026

WithAnyone: Toward Controllable and ID Consistent Image Generation

ICLR 2026poster

Identity-consistent (ID-consistent) generation has become an important focus in text-to-image research, with recent models achieving notable success in producing images aligned with a reference identity. Yet, the scarcity of large-scale paired datasets—containing multiple images of the same individu…

Cited by 0SourcecodeScholar
2026

iMontage: Unified, Versatile, Highly Dynamic Many-to-many Image Generation

CVPR 2026

Pre-trained video models learn powerful priors for generating high-quality, temporally coherent content. While these models excel at temporal coherence, their dynamics are often constrained by the continuous nature of their training data. We hypothesize that by injecting the rich and unconstrained c

Cited by 0SourcecodeScholar
2025

Beyond Sequences: Two-dimensional Representation and Dependency Encoding for Code Generation

ACL 2025long

The advent of large language models has significantly advanced automatic code generation, transforming the way programmers writing code. Inspired by natural language processing, mainstream code generation approaches represent code as a linear sequence of tokens. In this paper, we propose to represen…

Cited by 0SourcePDFScholar
2025

Chain-of-region: Visual Language Models Need Details for Diagram Analysis

ICLR 2025poster

Visual Language Models (VLMs) like GPT-4V have broadened the scope of LLM applications, yet they face significant challenges in accurately processing visual details, particularly in scientific diagrams. This paper explores the necessity of meticulous visual detail collection and region decompositio…

Cited by 1SourcePDFScholar
2025

DISC: Dynamic Decomposition Improves LLM Inference Scaling

NeurIPS 2025poster

Inference scaling methods for LLMs often rely on decomposing problems into steps (or groups of tokens), followed by sampling and selecting the best next steps. However, these steps and their sizes are often predetermined or manually designed based on domain knowledge. We propose dynamic decompositio…

Cited by 0SourceScholar
2025

F-Fidelity: A Robust Framework for Faithfulness Evaluation of Explainable AI

ICLR 2025poster

Recent research has developed a number of eXplainable AI (XAI) techniques, such as gradient-based approaches, input perturbation-base methods, and black-box explanation methods. While these XAI techniques can extract meaningful insights from deep learning models, how to properly evaluate them remain…

Cited by 0SourcePDFScholar
2025

FAVOR-Bench: A Comprehensive Benchmark for Fine-Grained Video Motion Understanding

NeurIPS 2025poster

Multimodal Large Language Models (MLLMs) have shown impressive video content understanding capabilities but struggle with fine-grained motion comprehension. To comprehensively assess the motion understanding ability of existing MLLMs, we introduce FAVOR-Bench, which comprises 1,776 videos from both…

Cited by 0SourceScholar
2025

Harnessing Vision Models for Time Series Analysis: A Survey

IJCAI 2025

Time series analysis has evolved from traditional autoregressive models to deep learning, Transformers, and Large Language Models (LLMs). While vision models have also been explored along the way, their contributions are less recognized due to the predominance of sequence modeling. However, challeng

2025

Human Texts Are Outliers: Detecting LLM-generated Texts via Out-of-distribution Detection

NeurIPS 2025poster

The rapid advancement of large language models (LLMs) such as ChatGPT, DeepSeek, and Claude has significantly increased the presence of AI-generated text in digital communication. This trend has heightened the need for reliable detection methods to distinguish between human-authored and machine-gene…

Cited by 0SourceScholar
2025

Humanizing the Machine: Proxy Attacks to Mislead LLM Detectors

ICLR 2025poster

The advent of large language models (LLMs) has revolutionized the field of text generation, producing outputs that closely mimic human-like writing. Although academic and industrial institutions have developed detectors to prevent the malicious usage of LLM-generated texts, other research has doubt…

Cited by 1SourcePDFScholar
2025

MixLLM: Dynamic Routing in Mixed Large Language Models

NAACL 2025long

Large Language Models (LLMs) exhibit potential artificial generic intelligence recently, however, their usage is costly with high response latency. Given mixed LLMs with their own strengths and weaknesses, LLM routing aims to identify the most suitable model for each query in the stream to maximize…

Cited by 4SourcePDFScholar
2025

Multi-Modal View Enhanced Large Vision Models for Long-Term Time Series Forecasting

NeurIPS 2025poster

Time series, typically represented as numerical sequences, can also be transformed into images and texts, offering multi-modal views (MMVs) of the same underlying signal. These MMVs can reveal complementary patterns and enable the use of powerful pre-trained large models, such as large vision models…

Cited by 0SourcecodeScholar
2025

OmniSVG: A Unified Scalable Vector Graphics Generation Model

NeurIPS 2025poster

Scalable Vector Graphics (SVG) is an important image format widely adopted in graphic design because of their resolution independence and editability. The study of generating high-quality SVG has continuously drawn attention from both designers and researchers in the AIGC community. However, existin…

Cited by 0SourceScholar
2025

OneIG-Bench: Omni-dimensional Nuanced Evaluation for Image Generation

NeurIPS 2025poster

Text-to-image (T2I) models have garnered significant attention for generating high-quality images aligned with text prompts. However, rapid T2I model advancements reveal limitations in early benchmarks, lacking comprehensive evaluations, especially for text rendering and style. Notably, recent state…

Cited by 0SourcecodeScholar
2025

Position Really Matters: Towards a Holistic Approach for Prompt Tuning

NAACL 2025findings

Prompt tuning is highly effective in efficiently extracting knowledge from foundation models, encompassing both language, vision, and vision-language models. However, the efficacy of employing fixed soft prompts with a predetermined position for concatenation with inputs for all instances, irrespect…

2025

SFS: Smarter Code Space Search improves LLM Inference Scaling

ICLR 2025poster

We frame code generation as a black-box optimization problem within the code space and demonstrate how optimization-inspired techniques can enhance inference scaling over text. Based on this perspective, we propose **SCATTERED FOREST SEARCH (SFS)**, a novel approach that improves solution diversity…

Cited by 0SourcePDFScholar
2025

SolverLLM: Leveraging Test-Time Scaling for Optimization Problem via LLM-Guided Search

NeurIPS 2025poster

Large Language Models (LLMs) offer promising capabilities for tackling complex reasoning tasks, including optimization problems. However, existing methods either rely on prompt engineering, which leads to poor generalization across problem types, or require costly supervised training. We introduce S…

Cited by 0SourceScholar
2025

Strategist: Self-improvement of LLM Decision Making via Bi-Level Tree Search

ICLR 2025poster

Traditional reinforcement learning and planning require a lot of data and training to develop effective strategies. On the other hand, large language models (LLMs) can generalize well and perform tasks without prior training but struggle with complex planning and decision-making. We introduce **STRA…

Cited by 0SourcePDFScholar
2025

TimeCAP: Learning to Contextualize, Augment, and Predict Time Series Events with Large Language Model Agents

AAAI 2025technical

Time series data is essential in various applications, including climate modeling, healthcare monitoring, and financial analytics. Understanding the contextual information associated with real-world time series data is often essential for accurate and reliable event predictions. In this paper, we in…

2025

TimeXL: Explainable Multi-modal Time Series Prediction with LLM-in-the-Loop

NeurIPS 2025poster

Time series analysis provides essential insights for real-world system dynamics and informs downstream decision-making, yet most existing methods often overlook the rich contextual signals present in auxiliary modalities. To bridge this gap, we introduce TimeXL, a multi-modal prediction framework th…

Cited by 0SourceScholar
2025

Where's the Liability in the Generative Era? Recovery-based Black-Box Detection of AI-Generated Content

CVPR 2025poster

The recent proliferation of photorealistic images created by generative models has sparked both excitement and concern, as these images are increasingly indistinguishable from real ones to the human eye. While offering new creative and commercial possibilities, the potential for misuse, such as in m…

Cited by 0SourcePDFScholar
2024

A Survey on Detection of LLMs-Generated Content

EMNLP 2024finding

The burgeoning capabilities of advanced large language models (LLMs) such as ChatGPT have led to an increase in synthetic content generation with implications across a variety of sectors, including media, cybersecurity, public discourse, and education. As such, the ability to detect LLMs-generated c…

2024

DALD: Improving Logits-based Detector without Logits from Black-box LLMs

NeurIPS 2024poster

The advent of Large Language Models (LLMs) has revolutionized text generation, producing outputs that closely mimic human writing. This blurring of lines between machine- and human-written text presents new challenges in distinguishing one from the other – a task further complicated by the frequent…

2024

DFA-RAG: Conversational Semantic Router for Large Language Model with Definite Finite Automaton

ICML 2024poster

This paper introduces the retrieval-augmented large language model with Definite Finite Automaton (DFA-RAG), a novel framework designed to enhance the capabilities of conversational agents using large language models (LLMs). Traditional LLMs face challenges in generating regulated and compliant resp…

Cited by 2SourcePDFScholar
2024

DNA-GPT: Divergent N-Gram Analysis for Training-Free Detection of GPT-Generated Text

ICLR 2024poster

Large language models (LLMs) have notably enhanced the fluency and diversity of machine-generated text. However, this progress also presents a significant challenge in detecting the origin of a given text, and current research on detection methods lags behind the rapid evolution of LLMs. Conventiona…

2024

Greenback Bears and Fiscal Hawks: Finance is a Jungle and Text Embeddings Must Adapt

EMNLP 2024industry

Financial documents are filled with specialized terminology, arcane jargon, and curious acronyms that pose challenges for general-purpose text embeddings. Yet, few text embeddings specialized for finance have been reported in the literature, perhaps in part due to a lack of public datasets and bench…

Cited by 3SourcePDFScholar
2024

InfuserKI: Enhancing Large Language Models with Knowledge Graphs via Infuser-Guided Knowledge Integration

EMNLP 2024finding

Large Language Models (LLMs) have achieved exceptional capabilities in open generation across various domains, yet they encounter difficulties with tasks that require intensive knowledge. To address these challenges, methods for integrating knowledge have been developed, which augment LLMs with doma…

Cited by 11SourcePDFScholar
2024

Large Language Models Can Be Contextual Privacy Protection Learners

EMNLP 2024main

The proliferation of Large Language Models (LLMs) has driven considerable interest in fine-tuning them with domain-specific data to create specialized language models. Nevertheless, such domain-specific fine-tuning data often contains contextually sensitive personally identifiable information (PII).…

2024

MeshXL: Neural Coordinate Field for Generative 3D Foundation Models

NeurIPS 2024poster

The polygon mesh representation of 3D data exhibits great flexibility, fast rendering speed, and storage efficiency, which is widely preferred in various applications. However, given its unstructured graph representation, the direct generation of high-fidelity 3D meshes is challenging. Fortunately,…

2024

Parametric Augmentation for Time Series Contrastive Learning

ICLR 2024poster

Modern techniques like contrastive learning have been effectively used in many areas, including computer vision, natural language processing, and graph-structured data. Creating positive examples that assist the model in learning robust and discriminative representations is a crucial stage in contra…

2024

Protecting Your LLMs with Information Bottleneck

NeurIPS 2024poster

The advent of large language models (LLMs) has revolutionized the field of natural language processing, yet they might be attacked to produce harmful content. Despite efforts to ethically align LLMs, these are often fragile and can be circumvented by jailbreaking attacks through optimized or manual…

2024

Pruning as a Domain-specific LLM Extractor

NAACL 2024findings

Large Language Models (LLMs) have exhibited remarkable proficiency across a wide array of NLP tasks. However, the escalation in model size also engenders substantial deployment costs. While few efforts have explored model pruning techniques to reduce the size of LLMs, they mainly center on general o…

2024

Towards Robust Fidelity for Evaluating Explainability of Graph Neural Networks

ICLR 2024poster

Graph Neural Networks (GNNs) are neural models that leverage the dependency structure in graphical data via message passing among the graph nodes. GNNs have emerged as pivotal architectures in analyzing graph-structured data, and their expansive application in sensitive domains requires a comprehens…

2024

TrustAgent: Towards Safe and Trustworthy LLM-based Agents

EMNLP 2024finding

The rise of LLM-based agents shows great potential to revolutionize task planning, capturing significant attention. Given that these agents will be integrated into high-stake domains, ensuring their reliability and safety is crucial. This paper presents an Agent-Constitution-based agent framework, T…

2024

Uncertainty Quantification for In-Context Learning of Large Language Models

NAACL 2024long

In-context learning has emerged as a groundbreaking ability of Large Language Models (LLMs) and revolutionized various fields by providing a few task-relevant demonstrations in the prompt. However, trustworthy issues with LLM’s response, such as hallucination, have also been actively discussed. Exis…

2023

DNA-Rendering: A Diverse Neural Actor Repository for High-Fidelity Human-Centric Rendering

ICCV 2023poster

Realistic human-centric rendering plays a key role in both computer vision and computer graphics. Rapid progress has been made in the algorithm aspect over the years, yet existing human-centric rendering datasets and benchmarks are rather impoverished in terms of diversity (e.g., outfit's fabric/mat…

Cited by 61PDFcodeScholar
2023

Hierarchical Gaussian Mixture based Task Generative Model for Robust Meta-Learning

NeurIPS 2023poster

Meta-learning enables quick adaptation of machine learning models to new tasks with limited data. While tasks could come from varying distributions in reality, most of the existing meta-learning methods consider both training and testing tasks as from the same uni-component distribution, overlooking…

Cited by 1SourcePDFScholar
2023

MonoHuman: Animatable Human Neural Field From Monocular Video

CVPR 2023poster

Animating virtual avatars with free-view control is crucial for various applications like virtual reality and digital entertainment. Previous studies have attempted to utilize the representation power of the neural radiance field (NeRF) to reconstruct the human body from monocular videos. Recent wor…

Cited by 94SourcePDFScholar
2023

Open-ended Commonsense Reasoning with Unrestricted Answer Candidates

EMNLP 2023long findings

Open-ended Commonsense Reasoning is defined as solving a commonsense question without providing 1) a short list of answer candidates and 2) a pre-defined answer scope. Conventional ways of formulating the commonsense question into a question-answering form or utilizing external knowledge to learn re…

Cited by 0SourceScholar
2023

Personalized Federated Learning under Mixture of Distributions

ICML 2023poster

The recent trend towards Personalized Federated Learning (PFL) has garnered significant attention as it allows for the training of models that are tailored to each client while maintaining data privacy. However, current PFL techniques primarily focus on modeling the conditional distribution heteroge…

2023

RenderMe-360: A Large Digital Asset Library and Benchmarks Towards High-fidelity Head Avatars

NeurIPS 2023poster

Synthesizing high-fidelity head avatars is a central problem for computer vision and graphics. While head avatar synthesis algorithms have advanced rapidly, the best ones still face great obstacles in real-world scenarios. One of the vital causes is the inadequate datasets -- 1) current public data…

2023

Time Series Contrastive Learning with Information-Aware Augmentations

AAAI 2023technical

Various contrastive learning approaches have been proposed in recent years and achieve significant empirical success. While effective and prevalent, contrastive learning has been less explored for time series data. A key component of contrastive learning is to select appropriate augmentations imposi…

2023

Towards Robust Pruning: An Adaptive Knowledge-Retention Pruning Strategy for Language Models

EMNLP 2023long main

The pruning objective has recently extended beyond accuracy and sparsity to robustness in language models. Despite this, existing methods struggle to enhance robustness against adversarial attacks when continually increasing model sparsity and require a retraining process. As humans step into the er…

Cited by 0SourceScholar
2022

DHWP: Learning High-Quality Short Hash Codes Via Weight Pruning

ICASSP 2022accepted

Hashing is widely used in large-scale image retrieval because of its efficiency in storage and computation. Although longer hash codes can lead to higher search accuracy, the retrieval cost increases linearly with the increase of the number of hash bits. Most deep hashing methods suffer from the pro…

Cited by 0SourceScholar
2022

Seed: Sound Event Early Detection Via Evidential Uncertainty

ICASSP 2022accepted

Sound Event Early Detection (SEED) is an essential task in recognizing the acoustic environments and soundscapes. However, most of the existing methods focus on the offline sound event detection, which suffers from the over-confidence issue of early-stage event detection and usually yield unreliable…

Cited by 0SourceScholar
2022

Superclass-Conditional Gaussian Mixture Model For Learning Fine-Grained Embeddings

ICLR 2022spotlight

Learning fine-grained embeddings is essential for extending the generalizability of models pre-trained on "coarse" labels (e.g., animals). It is crucial to fields for which fine-grained labeling (e.g., breeds of animals) is expensive, but fine-grained prediction is desirable, such as medicine. The d…

2022

Zero-Shot Cross-Lingual Machine Reading Comprehension via Inter-sentence Dependency Graph

AAAI 2022technical

We target the task of cross-lingual Machine Reading Comprehension (MRC) in the direct zero-shot setting, by incorporating syntactic features from Universal Dependencies (UD), and the key features we use are the syntactic relations within each sentence. While previous work has demonstrated effective…

2021

Adaptive Prior-Dependent Correction Enhanced Reinforcement Learning for Natural Language Generation

AAAI 2021technical

Natural language generation (NLG) is an important task with various applications like neural machine translation (NMT) and image captioning. Since deep-learning-based methods have issues of exposure bias and loss inconsistency, reinforcement learning (RL) is widely adopted in NLG tasks recently. But…

Cited by 2SourcePDFScholar
2021

Dynamic Gaussian Mixture based Deep Generative Model For Robust Forecasting on Sparse Multivariate Time Series

AAAI 2021technical

Forecasting on sparse multivariate time series (MTS) aims to model the predictors of future values of time series given their incomplete past, which is important for many emerging applications. However, most existing methods process MTS’s individually, and do not leverage the dynamic distributions u…

2021

FaceSec: A Fine-Grained Robustness Evaluation Framework for Face Recognition Systems

CVPR 2021poster

We present FACESEC, a framework for fine-grained robustness evaluation of face recognition systems. FACESEC evaluation is performed along four dimensions of adversarial modeling: the nature of perturbation (e.g., pixel-level or face accessories), the attacker's system knowledge (about training data…

Cited by 27PDFcodeScholar
2021

InfoGCL: Information-Aware Graph Contrastive Learning

NeurIPS 2021poster

Various graph contrastive learning models have been proposed to improve the performance of tasks on graph datasets in recent years. While effective and prevalent, these models are usually carefully customized. In particular, despite all recent work create two contrastive views, they differ in a vari…

Cited by 235SourcePDFScholar
2021

Recommend for a Reason: Unlocking the Power of Unsupervised Aspect-Sentiment Co-Extraction

EMNLP 2021finding

Compliments and concerns in reviews are valuable for understanding users’ shopping interests and their opinions with respect to specific aspects of certain items. Existing review-based recommenders favor large and complex language encoders that can only learn latent and uninterpretable text represen…

2021

Transformer-Style Relational Reasoning with Dynamic Memory Updating for Temporal Network Modeling

AAAI 2021technical

Network modeling aims to learn the latent representations of nodes such that the representations preserve both network structures and node attribute information. This problem is fundamental due to its prevalence in numerous domains. However, existing approaches either target the static networks or s…

Cited by 24SourcePDFScholar
2021

Unsupervised Concept Representation Learning for Length-Varying Text Similarity

NAACL 2021long

Measuring document similarity plays an important role in natural language processing tasks. Most existing document similarity approaches suffer from the information gap caused by context and vocabulary mismatches when comparing varying-length texts. In this paper, we propose an unsupervised concept…

2020

Inductive and Unsupervised Representation Learning on Graph Structured Objects

ICLR 2020poster

Inductive and unsupervised graph learning is a critical technique for predictive or information retrieval tasks where label information is difficult to obtain. It is also challenging to make graph learning inductive and unsupervised at the same time, as learning processes guided by reconstruction er…

Cited by 34SourceScholar
2020

Parameterized Explainer for Graph Neural Network

NeurIPS 2020poster

Despite recent progress in Graph Neural Networks (GNNs), explaining predictions made by GNNs remains a challenging open problem. The leading method mainly addresses the local explanations (i.e., important subgraph structure and node features) to interpret why a GNN model makes the prediction for a s…

2020

Robust Graph Representation Learning via Neural Sparsification

ICML 2020poster

Graph representation learning serves as the core of important prediction tasks, ranging from product recommendation to fraud detection. Real-life graphs usually have complex information in the local neighborhood, where each node is described by a rich set of features and connects to dozens or even h…

Cited by 366SourcePDFScholar
2018

Deep Autoencoding Gaussian Mixture Model for Unsupervised Anomaly Detection

ICLR 2018poster

Unsupervised anomaly detection on multi- or high-dimensional data is of great importance in both fundamental machine learning research and industrial applications, for which density estimation lies at the core. Although previous approaches based on dimensionality reduction followed by density estima…

Cited by 2410SourcePDFScholar