← Search

Shaohan Huang

54 accepted papers

2026

AesCoder: Code Aesthetics with Agentic Reward Feedback

ICLR 2026poster

Large Language Models (LLMs) have become valuable assistants for developers in code-related tasks. While LLMs excel at traditional programming tasks such as code generation and bug fixing, they struggle with visually-oriented coding tasks, often producing suboptimal aesthetics. In this paper, we int…

Cited by 0SourcecodeScholar
2026

Multimodal Latent Language Modeling with Next-Token Diffusion

ICML 2026spotlight

Multimodal generative models require a unified approach to handle both discrete data (e.g., text and code) and continuous data (e.g., image, audio, video). In this work, we propose Latent Language Modeling (LatentLM), which seamlessly integrates continuous and discrete data using causal Transformers…

Cited by 0SourcecodeScholar
2026

Reasoning with Exploration: An Entropy Perspective

AAAI 2026technical

Balancing exploration and exploitation is a central goal in reinforcement learning (RL). Despite recent advances in enhancing language model (LM) reasoning, most methods lean toward exploitation, and increasingly encounter performance plateaus. In this work, we revisit entropy -- a signal of explora

Cited by 0SourcePDFScholar
2026

SlideSparse: Fast and Flexible (2N-2):2N Structured Sparsity

ICML 2026poster

NVIDIA's 2:4 Sparse Tensor Cores deliver $2\times$ throughput but demand 50% pruning—a ratio that collapses LLM reasoning accuracy (Qwen3: 54%→15%). Milder $(2N-2):2N$ patterns (e.g., 6:8, 25% pruning) preserve accuracy yet receive *no* hardware support, falling back to dense execution. We present *…

Cited by 0SourceScholar
2026

VibeVoice: Expressive Podcast Generation with Next-Token Diffusion

ICLR 2026oral

Generating long-form, multi-speaker conversational audio like podcasts poses significant challenges for traditional Text-to-Speech (TTS) systems, particularly in scalability, speaker consistency, and natural turn-taking. We present VibeVoice , a novel model designed to synthesize expressive, long-fo…

Cited by 0SourceScholar
2026

VisCodex: Unified Multimodal Code Generation via Merging Vision and Coding Models

ICLR 2026poster

Multimodal large language models (MLLMs) have significantly advanced the integration of visual and textual understanding. However, their ability to generate code from multimodal inputs remains limited. In this work, we introduce VisCodex, a unified framework that seamlessly merges vision and coding…

Cited by 0SourcecodeScholar
2025

Context-DPO: Aligning Language Models for Context-Faithfulness

ACL 2025finding

Reliable responses from large language models (LLMs) require adherence to user instructions and retrieved information. While alignment techniques help LLMs align with human intentions and values, improving context-faithfulness through alignment remains underexplored. To address this, we propose Cont…

2025

GeAR: Generation Augmented Retrieval

ACL 2025finding

Document retrieval techniques are essential for developing large-scale information systems. The common approach involves using a bi-encoder to compute the semantic similarity between a query and documents. However, the scalar similarity often fail to reflect enough information, hindering the interpr…

2025

NL2Lean: Translating Natural Language into Lean 4 through Multi-Aspect Reinforcement Learning

EMNLP 2025

Translating natural language into formal language such as Lean 4 has gained attention for its potential to automate formal proof development. Automated methods provide a scalable and cost-effective alternative to manual formalization, driving increasing interest in this task. However, existing LLMs

Cited by 0SourcePDFScholar
2025

On Domain-Adaptive Post-Training for Multimodal Large Language Models

EMNLP 2025

Adapting general multimodal large language models (MLLMs) to specific domains, such as scientific and industrial fields, is highly significant in promoting their practical applications. This paper systematically investigates domain adaptation of MLLMs via post-training, focusing on data synthesis, t

Cited by 0SourcePDFScholar
2025

Think Only When You Need with Large Hybrid-Reasoning Models

NeurIPS 2025poster

Recent Large Reasoning Models (LRMs) have shown substantially improved reasoning capabilities over traditional Large Language Models (LLMs) by incorporating extended thinking processes prior to producing final responses. However, excessively lengthy thinking introduces substantial overhead in terms…

Cited by 0SourceScholar
2024

Boosting Text-to-Video Generative Model with MLLMs Feedback

NeurIPS 2024poster

Recent advancements in text-to-video generative models, such as Sora, have showcased impressive capabilities. These models have attracted significant interest for their potential applications. However, they often rely on extensive datasets of variable quality, which can result in generated videos th…

Cited by 5SourcePDFScholar
2024

Calibrating LLM-Based Evaluator

COLING 2024main

Recent advancements in large language models (LLMs) and their emergent capabilities make LLM a promising reference-free evaluator on the quality of natural language generation, and a competent alternative to human evaluation. However, hindered by the closed-source or high computational demand to hos…

Cited by 73SourcePDFScholar
2024

Grounding Multimodal Large Language Models to the World

ICLR 2024poster

We introduce Kosmos-2, a Multimodal Large Language Model (MLLM), enabling new capabilities of perceiving object descriptions (e.g., bounding boxes) and grounding text to the visual world. Specifically, we represent text spans (i.e., referring expressions and noun phrases) as links in Markdown, i.e.,…

Cited by 54SourcePDFScholar
2024

HD-Eval: Aligning Large Language Model Evaluators Through Hierarchical Criteria Decomposition

ACL 2024long

Large language models (LLMs) have emerged as a promising alternative to expensive human evaluations. However, the alignment and coverage of LLM-based evaluations are often limited by the scope and potential bias of the evaluation prompts and criteria. To address this challenge, we propose HD-Eval, a…

2024

Instruction Pre-Training: Language Models are Supervised Multitask Learners

EMNLP 2024main

Unsupervised multitask pre-training has been the critical method behind the recent success of language models (LMs). However, supervised multitask learning still holds significant promise, as scaling it in the post-training stage trends towards better generalization. In this paper, we explore superv…

2024

KOSMOS-E : Learning to Follow Instruction for Robotic Grasping

IROS 2024poster

Tuning on instruction-following data has been shown to enhance the capabilities and controllability of language models, but the idea is less explored in the robotic field. In this work, we introduce KOSMOS-E, a Multimodal Large Language Model (MLLM) that leverages instruction-following robotic grasp…

Cited by 1SourceScholar
2024

Kosmos-G: Generating Images in Context with Multimodal Large Language Models

ICLR 2024poster

Recent advancements in subject-driven image generation have made significant strides. However, current methods still fall short in diverse application scenarios, as they require test-time tuning and cannot accept interleaved multi-image and text input. These limitations keep them far from the ultima…

2024

Multimodal Large Language Models Make Text-to-Image Generative Models Align Better

NeurIPS 2024poster

Recent studies have demonstrated the exceptional potentials of leveraging human preference datasets to refine text-to-image generative models, enhancing the alignment between generated images and textual prompts. Despite these advances, current human preference datasets are either prohibitively expe…

Cited by 3SourcePDFScholar
2024

ResLoRA: Identity Residual Mapping in Low-Rank Adaption

ACL 2024findings

As one of the most popular parameter-efficient fine-tuning (PEFT) methods, low-rank adaptation (LoRA) is commonly applied to fine-tune large language models (LLMs). However, updating the weights of LoRA blocks effectively and expeditiously is challenging due to the long calculation path in the origi…

2024

Scaling Sentence Embeddings with Large Language Models

EMNLP 2024finding

Large Language Models (LLMs) have recently gained significant interest due to their impressive results in various natural language tasks. However, their application to sentence embeddings is still under active research. In this work, we introduce PromptEOL, a simple and efficient method designed to…

2024

Se2: Sequential Example Selection for In-Context Learning

ACL 2024findings

The remarkable capability of large language models(LLMs) for in-context learning(ICL) needs to be activated by demonstration examples. Prior work has extensively explored the selection of examples for ICL, predominantly following the “select then organize” paradigm, such approaches often neglect the…

2024

Text Diffusion with Reinforced Conditioning

AAAI 2024technical

Diffusion models have demonstrated exceptional capability in generating high-quality images, videos, and audio. Due to their adaptiveness in iterative refinement, they provide a strong potential for achieving better non-autoregressive sequence generation. However, existing text diffusion models stil…

Cited by 1SourcePDFScholar
2024

You Only Cache Once: Decoder-Decoder Architectures for Language Models

NeurIPS 2024oral

We introduce a decoder-decoder architecture, YOCO, for large language models, which only caches key-value pairs once. It consists of two components, i.e., a cross-decoder stacked upon a self-decoder. The self-decoder efficiently encodes global key-value (KV) caches that are reused by the cross-decod…

2023

A Length-Extrapolatable Transformer

ACL 2023long

Position modeling plays a critical role in Transformers. In this paper, we focus on length extrapolation, i.e., training on short texts while evaluating longer sequences. We define attention resolution as an indicator of extrapolation. Then we propose two designs to improve the above metric of Trans…

2023

Beyond English-Centric Bitexts for Better Multilingual Language Representation Learning

ACL 2023long

In this paper, we elaborate upon recipes for building multilingual representation models that are not only competitive with existing state-of-the-art models but are also more parameter efficient, thereby promoting better adoption in resource-constrained scenarios and practical applications. We show…

Cited by 19SourcePDFScholar
2023

Democratizing Reasoning Ability: Tailored Learning from Large Language Model

EMNLP 2023long main

Large language models (LLMs) exhibit impressive emergent abilities in natural language processing, but their democratization is hindered due to huge computation requirements and closed-source nature. Recent research on advancing open-source smaller LMs by distilling knowledge from black-box LLMs has…

Cited by 0SourcecodeScholar
2023

Dual-Alignment Pre-training for Cross-lingual Sentence Embedding

ACL 2023long

Recent studies have shown that dual encoder models trained with the sentence-level translation ranking task are effective methods for cross-lingual sentence embedding. However, our research indicates that token-level alignment is also crucial in multilingual scenarios, which has not been fully explo…

2023

GanLM: Encoder-Decoder Pre-training with an Auxiliary Discriminator

ACL 2023long

Pre-trained models have achieved remarkable success in natural language processing (NLP). However, existing pre-training methods underutilize the benefits of language understanding for generation. Inspired by the idea of Generative Adversarial Networks (GANs), we propose a GAN-style model for encode…

2023

Language Is Not All You Need: Aligning Perception with Language Models

NeurIPS 2023poster

A big convergence of language, multimodal perception, action, and world modeling is a key step toward artificial general intelligence. In this work, we introduce KOSMOS-1, a Multimodal Large Language Model (MLLM) that can perceive general modalities, learn in context (i.e., few-shot), and follow ins…

2023

Magneto: A Foundation Transformer

ICML 2023poster

A big convergence of model architectures across language, vision, speech, and multimodal is emerging. However, under the same name ''Transformers'', the above areas use different implementations for better performance, e.g., Post-LayerNorm for BERT, and Pre-LayerNorm for GPT and vision Transformers.…

Cited by 12SourcePDFScholar
2023

Pre-training Language Model as a Multi-perspective Course Learner

ACL 2023findings

ELECTRA, the generator-discriminator pre-training framework, has achieved impressive semantic construction capability among various downstream tasks. Despite the convincing performance, ELECTRA still faces the challenges of monotonous training and deficient interaction. Generator with only masked la…

Cited by 1SourcePDFScholar
2023

UPRISE: Universal Prompt Retrieval for Improving Zero-Shot Evaluation

EMNLP 2023long main

Large Language Models (LLMs) are popular for their impressive abilities, but the need for model-specific fine-tuning or task-specific prompt engineering can hinder their generalization. We propose UPRISE (Universal Prompt Retrieval for Improving zero-Shot Evaluation), which tunes a lightweight and v…

Cited by 0SourcecodeScholar
2022

CROP: Zero-shot Cross-lingual Named Entity Recognition with Multilingual Labeled Sequence Translation

EMNLP 2022finding

Named entity recognition (NER) suffers from the scarcity of annotated training data, especially for low-resource languages without labeled data. Cross-lingual NER has been proposed to alleviate this issue by transferring knowledge from high-resource languages to low-resource languages via aligned cr…

2022

Learning Music Sequence Representation From Text Supervision

ICASSP 2022accepted

Music representation learning is notoriously difficult for its complex human-related concepts contained in the sequence of numerical signals. To excavate better MUsic SEquence Representation from labeled audio, we propose a novel text-supervision pre-training method, namely MUSER. MUSER adopts an au…

Cited by 0SourceScholar
2022

On the Representation Collapse of Sparse Mixture of Experts

NeurIPS 2022accept

Sparse mixture of experts provides larger model capacity while requiring a constant computational overhead. It employs the routing mechanism to distribute input tokens to the best-matched experts according to their hidden representations. However, learning such a routing mechanism encourages token c…

2022

PromptBERT: Improving BERT Sentence Embeddings with Prompts

EMNLP 2022main

We propose PromptBERT, a novel contrastive learning method for learning better sentence representation. We firstly analysis the drawback of current sentence embedding from original BERT and find that it is mainly due to the static token embedding bias and ineffective BERT layers. Then we propose the…

2022

Snapshot-Guided Domain Adaptation for ELECTRA

EMNLP 2022finding

Discriminative pre-trained language models, such as ELECTRA, have achieved promising performances in a variety of general tasks. However, these generic pre-trained models struggle to capture domain-specific knowledge of domain-related tasks. In this work, we propose a novel domain-adaptation method…

2022

THE-X: Privacy-Preserving Transformer Inference with Homomorphic Encryption

ACL 2022findings

As more and more pre-trained language models adopt on-cloud deployment, the privacy issues grow quickly, mainly for the exposure of plain-text user data (e.g., search history, medical record, bank account). Privacy-preserving inference of transformer models is on the demand of cloud service users. T…

2022

XLM-E: Cross-lingual Language Model Pre-training via ELECTRA

ACL 2022long

In this paper, we introduce ELECTRA-style tasks to cross-lingual language model pre-training. Specifically, we present two pre-training tasks, namely multilingual replaced token detection, and translation replaced token detection. Besides, we pretrain the model, named as XLM-E, on both multilingual…

2021

Allocating Large Vocabulary Capacity for Cross-Lingual Language Model Pre-Training

EMNLP 2021main

Compared to monolingual models, cross-lingual models usually require a more expressive vocabulary to represent all languages adequately. We find that many languages are under-represented in recent cross-lingual language models due to the limited vocabulary capacity. To this end, we propose an algori…

2021

Consistency Regularization for Cross-Lingual Fine-Tuning

ACL 2021long

Fine-tuning pre-trained cross-lingual language models can transfer task-specific supervision from one language to the others. In this work, we propose to improve cross-lingual fine-tuning with consistency regularization. Specifically, we use example consistency regularization to penalize the predict…

2021

Improving Pretrained Cross-Lingual Language Models via Self-Labeled Word Alignment

ACL 2021long

The cross-lingual language models are typically pretrained with masked language modeling on multilingual text or parallel sentences. In this paper, we introduce denoising word alignment as a new cross-lingual pre-training task. Specifically, the model first self-label word alignments for parallel se…

2021

mT6: Multilingual Pretrained Text-to-Text Transformer with Translation Pairs

EMNLP 2021main

Multilingual T5 pretrains a sequence-to-sequence model on massive monolingual texts, which has shown promising results on many cross-lingual tasks. In this paper, we improve multilingual text-to-text transfer Transformer with translation pairs (mT6). Specifically, we explore three cross-lingual text…

2020

DocBank: A Benchmark Dataset for Document Layout Analysis

COLING 2020main

Document layout analysis usually relies on computer vision models to understand documents while ignoring textual information that is vital to capture. Meanwhile, high quality labeled datasets with both visual and textual information are still insufficient. In this paper, we present DocBank, a benchm…