← Search

Jiaheng Wei

19 accepted papers

2026

Calibrated Multimodal Representation Learning with Missing Modalities

ICML 2026poster

Multimodal representation learning harmonizes distinct modalities by aligning them into a unified latent space. Recent research generalizes traditional cross-modal alignment to produce enhanced multimodal synergy but requires all modalities to be present for a common instance, making it challenging …

Cited by 0SourceScholar
2026

Detecting Errors in AI-Generated Annotations: When and Why Semantic Neighbors Help

ICML 2026poster

Large language models (LLMs) and vision-language models (VLMs) have emerged as efficient annotators for tasks such as generation and classification. While these models offer significant cost and speed advantages over human annotation, a critical challenge remains: existing self-evaluation methods, s…

Cited by 0SourceScholar
2026

Reassessing Layer Pruning in LLMs: New Insights and Methods

ICLR 2026poster

Although large language models (LLMs) have achieved remarkable success across various domains, their considerable scale necessitates substantial computational resources, posing significant challenges for deployment in resource-constrained environments. Layer pruning, as a simple yet effective compre…

Cited by 0SourcecodeScholar
2026

Robust Learning from Noisily Labeled Long-Tailed Data via Fairness Regularizer

AAAI 2026technical

Both long-tailed and noisily labeled data frequently appear in real-world applications and impose significant challenges for learning. Most prior works treat either problem in an isolated way and do not explicitly consider the coupling effects of the two. Our empirical observation reveals that such

Cited by 0SourcePDFScholar
2026

Task-Aware Data Selection via Proxy-Label Enhanced Distribution Matching for LLM Finetuning

ICLR 2026poster

Task-specific fine-tuning of foundation models is critically dependent on the quality and relevance of the instruction data. While prevailing data selection methods rely exclusively on instruction instances X to approximate the target distribution, we argue that selection should align with the joint…

Cited by 0SourceScholar
2026

VP-Bench: A Comprehensive Benchmark for Visual Prompting in Multimodal Large Language Models

AAAI 2026technical

Multimodal Large Language Models (MLLM) have enabled a wide range of advanced vision-language applications, including fine-grained object recognition and contextual understanding. When querying specific regions or objects in an image, human users naturally use "Visual Prompts" (VP) like bounding box

Cited by 0SourcePDFScholar
2025

Human and AI Perceptual Differences in Image Classification Errors

AAAI 2025technical

Artificial intelligence (AI) models for computer vision trained with supervised machine learning are assumed to solve classification tasks by imitating human behavior learned from training labels. Most efforts in recent vision research focus on measuring the model task performance using standardized…

Cited by 0SourcePDFScholar
2025

Improving Data Efficiency via Curating LLM-Driven Rating Systems

ICLR 2025poster

Instruction tuning is critical for adapting large language models (LLMs) to downstream tasks, and recent studies have demonstrated that small amounts of human-curated data can outperform larger datasets, challenging traditional data scaling laws. While LLM-based data quality rating systems offer a c…

Cited by 3SourcePDFScholar
2025

LLM Unlearning via Loss Adjustment with Only Forget Data

ICLR 2025poster

Unlearning in Large Language Models (LLMs) is essential for ensuring ethical and responsible AI use, especially in addressing privacy leak, bias, safety, and evolving regulations. Existing approaches to LLM unlearning often rely on retain data or a reference LLM, yet they struggle to adequately bala…

Cited by 2SourcePDFScholar
2025

RULE: Reinforcement UnLEarning Achieves Forget-retain Pareto Optimality

NeurIPS 2025poster

The widespread deployment of Large Language Models (LLMs) trained on massive, uncurated corpora has raised growing concerns about the inclusion of sensitive, copyrighted, or illegal content. This has led to increasing interest in LLM unlearning: the task of selectively removing specific information…

Cited by 0SourceScholar
2025

Recognition through Reasoning: Reinforcing Image Geo-localization with Large Vision-Language Models

NeurIPS 2025poster

Previous methods for image geo-localization have typically treated the task as either classification or retrieval, often relying on black-box decisions that lack interpretability. The rise of large vision-language models (LVLMs) has enabled a rethinking of geo-localization as a reasoning-driven task…

Cited by 0SourcecodeScholar
2025

Token Cleaning: Fine-Grained Data Selection for LLM Supervised Fine-Tuning

ICML 2025poster

Recent studies show that in supervised fine-tuning (SFT) of large language models (LLMs), data quality matters more than quantity. While most data cleaning methods concentrate on filtering entire samples, the quality of individual tokens within a sample can vary significantly. After pre-training, e…

2023

Distributionally Robust Post-hoc Classifiers under Prior Shifts

ICLR 2023poster

The generalization ability of machine learning models degrades significantly when the test distribution shifts away from the training distribution. We investigate the problem of training models that are robust to shifts caused by changes in the distribution of class-priors or group-priors. The prese…

2022

DuelGAN: A Duel between Two Discriminators Stabilizes the GAN Training

ECCV 2022poster

"In this paper, we introduce DuelGAN, a generative adversarial network (GAN) solution to improve the stability of the generated samples and to mitigate mode collapse. Built upon the Vanilla GAN’s two-player game between the discriminator D_1 and the generator G, we introduce a peer discriminator D_2…

2022

Learning with Noisy Labels Revisited: A Study Using Real-World Human Annotations

ICLR 2022poster

Existing research on learning with noisy labels mainly focuses on synthetic label noise. The synthetic noise, though has clean structures which greatly enabled statistical analyses, often fails to model the real-world noise patterns. The recent literature has observed several efforts to offer real-w…

2022

To Smooth or Not? When Label Smoothing Meets Noisy Labels

ICML 2022oral

Label smoothing (LS) is an arising learning paradigm that uses the positively weighted average of both the hard training labels and uniformly distributed soft labels. It was shown that LS serves as a regularizer for training data with hard labels and therefore improves the generalization of the mode…