← Search

Xuezhe Ma

28 accepted papers

2026

Are VLMs Seeing or Just Saying? Uncovering the Illusion of Visual Re-examination

ICML 2026oral

Vision-Language Models (VLMs) frequently generate self-reflective statements during reasoning, such as ``let me check the figure again.'' Do such statements trigger genuine visual re-examination, or merely represent learned textual patterns? We investigate this question through VisualSwap, an image-…

Cited by 3SourceScholar
2026

MegaCoin: Enhancing Medium-Grained Color Perception for Vision-Language Models

AAAI 2026technical

In vision-language models (VLMs), the ability to perceive and interpret color and physical environment is crucial for achieving contextually accurate understanding and interaction. However, despite advances in multimodal modeling, there remains a significant lack of specialized datasets that rigorou

Cited by 0SourcePDFScholar
2025

LLM The Genius Paradox: A Linguistic and Math Expert’s Struggle with Simple Word-based Counting Problems

NAACL 2025long

Interestingly, LLMs yet struggle with some basic tasks that humans find trivial to handle, e.g., counting the number of character r’s in the word “strawberry”. There are several popular conjectures (e.g., tokenization, architecture and training data) regarding the reason for deficiency of LLMs in si…

2025

Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model

ICLR 2025oral

We introduce Transfusion, a recipe for training a multi-modal model over discrete and continuous data. Transfusion combines the language modeling loss function (next token prediction) with diffusion to train a single transformer over mixed-modality sequences. We pretrain multiple Transfusion models…

Cited by 150SourcePDFScholar
2024

Light-weight Fine-tuning Method for Defending Adversarial Noise in Pre-trained Medical Vision-Language Models

EMNLP 2024finding

Fine-tuning pre-trained Vision-Language Models (VLMs) has shown remarkable capabilities in medical image and textual depiction synergy. Nevertheless, many pre-training datasets are restricted by patient privacy concerns, potentially containing noise that can adversely affect downstream performance.…

Cited by 2SourcePDFScholar
2024

MIDDAG: Where Does Our News Go? Investigating Information Diffusion via Community-Level Information Pathways

AAAI 2024technical

We present MIDDAG, an intuitive, interactive system that visualizes the information propagation paths on social media triggered by COVID-19-related news articles accompanied by comprehensive insights including user/community susceptibility level, as well as events and popular opinions raised by the…

2024

Megalodon: Efficient LLM Pretraining and Inference with Unlimited Context Length

NeurIPS 2024poster

The quadratic complexity and weak length extrapolation of Transformers limits their ability to scale to long sequences, and while sub-quadratic solutions like linear attention and state space models exist, they empirically underperform Transformers in pretraining efficiency and downstream task accur…

2023

Better May Not Be Fairer: A Study on Subgroup Discrepancy in Image Classification

ICCV 2023poster

In this paper, we provide 20,000 non-trivial human annotations on popular datasets as a first step to bridge gap to studying how natural semantic spurious features affect image classification, as prior works often study datasets mixing low-level features due to limitations in accessing realistic dat…

Cited by 5PDFcodeScholar
2023

Evaluating Large Language Models on Controlled Generation Tasks

EMNLP 2023long main

While recent studies have looked into the abilities of large language models in various benchmark tasks, including question generation, reading comprehension, multilingual and etc, there have been few studies looking into the controllability of large language models on generation tasks. We present a…

Cited by 0SourcecodeScholar
2023

Know Where You’re Going: Meta-Learning for Parameter-Efficient Fine-Tuning

ACL 2023findings

A recent family of techniques, dubbed lightweight fine-tuning methods, facilitates parameter-efficient transfer by updating only a small set of additional parameters while keeping the parameters of the original model frozen. While proven to be an effective approach, there are no existing studies on…

Cited by 11SourcePDFScholar
2023

Mega: Moving Average Equipped Gated Attention

ICLR 2023poster

The design choices in the Transformer attention mechanism, including weak inductive bias and quadratic computational complexity, have limited its application for modeling long sequences. In this paper, we introduce Mega, a simple, theoretically grounded, single-head gated attention mechanism equippe…

2023

RECAP: Retrieval-Enhanced Context-Aware Prefix Encoder for Personalized Dialogue Response Generation

ACL 2023long

Endowing chatbots with a consistent persona is essential to an engaging conversation, yet it remains an unresolved challenge. In this work, we propose a new retrieval-enhanced approach for personalized response generation. Specifically, we design a hierarchical transformer retriever trained on dialo…

2022

Improving Stability of Fine-Tuning Pretrained Language Models via Component-Wise Gradient Norm Clipping

EMNLP 2022main

Fine-tuning over large pretrained language models (PLMs) has established many state-of-the-art results. Despite its superior performance, such fine-tuning can be unstable, resulting in significant variance in performance and potential risks for practical applications. Previous works have attributed…

2022

Prompt Consistency for Zero-Shot Task Generalization

EMNLP 2022finding

One of the most impressive results of recent NLP history is the ability of pre-trained language models to solve new tasks in a zero-shot setting. To achieve this, NLP tasks are framed as natural language prompts, generating a response indicating the predicted output. Nonetheless, the performance in…

2022

Towards a Unified View of Parameter-Efficient Transfer Learning

ICLR 2022spotlight

Fine-tuning large pretrained language models on downstream tasks has become the de-facto learning paradigm in NLP. However, conventional approaches fine-tune all the parameters of the pretrained model, which becomes prohibitive as the model size and the number of tasks grow. Recent work has proposed…

2021

Decoupling Global and Local Representations via Invertible Generative Flows

ICLR 2021poster

In this work, we propose a new generative model that is capable of automatically decoupling global and local representations of images in an entirely unsupervised setting, by embedding a generative flow in the VAE framework to model the decoder. Specifically, the proposed model utilizes the variatio…

2021

DiSCoL: Toward Engaging Dialogue Systems through Conversational Line Guided Response Generation

NAACL 2021system demonstrations

Having engaging and informative conversations with users is the utmost goal for open-domain conversational systems. Recent advances in transformer-based language models and their applications to dialogue systems have succeeded to generate fluent and human-like responses. However, they still lack con…

Cited by 14SourcePDFScholar
2021

Examining and Combating Spurious Features under Distribution Shift

ICML 2021spotlight

A central goal of machine learning is to learn robust representations that capture the fundamental relationship between inputs and output labels. However, minimizing training errors over finite or biased datasets results in models latching on to spurious correlations between the training input/outpu…

2021

Luna: Linear Unified Nested Attention

NeurIPS 2021poster

The quadratic computational and memory complexities of the Transformer's attention mechanism have limited its scalability for modeling long sequences. In this paper, we propose Luna, a linear unified nested attention mechanism that approximates softmax attention with two nested linear attention fun…

2019

MAE: Mutual Posterior-Divergence Regularization for Variational AutoEncoders

ICLR 2019poster

Variational Autoencoder (VAE), a simple and effective deep generative model, has led to a number of impressive empirical successes and spawned many advanced variants and theoretical investigations. However, recent studies demonstrate that, when equipped with expressive generative distributions (aka.…

Cited by 44SourcePDFScholar
2017

Dropout with Expectation-linear Regularization

ICLR 2017poster

Dropout, a simple and effective way to train deep neural networks, has led to a number of impressive empirical successes and spawned many recent theoretical investigations. However, the gap between dropout’s training and inference phases, introduced due to tractability considerations, has largely re…

Cited by 61SourceScholar