← Search

Lei Cui

32 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

FormAct: Agentic Source Editing for Rich-Format Document Generation

ICML 2026poster

Rich-format documents are essential for everyday operations yet costly to author, motivating the need for automated generation to enhance productivity. To this end, we present FormAct, an agentic system that generates professional rich-format documents from scratch. FormAct operates on an HTML sourc…

Cited by 0SourceScholar
2026

Synergizing Understanding and Generation with Interleaved Analyzing-Drafting Thinking

ICLR 2026poster

Unified Vision–Language Models (UVLMs) aim to advance multimodal learning by supporting both understanding and generation within a single framework. However, existing approaches largely focus on architectural unification while overlooking the need for explicit interaction between the two capabilitie…

Cited by 0SourceScholar
2026

Towards Effective and Efficient Context-aware Nucleus Detection in Histopathology Whole Slide Images

AAAI 2026technical

Nucleus detection in histopathology whole slide images (WSIs) is crucial for a broad spectrum of clinical applications. The gigapixel size of WSIs necessitates the use of sliding window methodology for nucleus detection. However, mainstream methods process each sliding window independently, which ov

Cited by 0SourcePDFScholar
2026

UniM: A Unified Any-to-Any Interleaved Multimodal Benchmark

CVPR 2026

In real-world multimodal applications, systems usually need to comprehend arbitrarily combined and interleaved multimodal inputs from users, while also generating outputs in any interleaved multimedia form. This capability defines the goal of any-to-any interleaved multimodal learning under a unifie

Cited by 0SourceScholar
2025

CLeVeR: Multi-modal Contrastive Learning for Vulnerability Code Representation

ACL 2025finding

Automated vulnerability detection has become increasingly important. Many existing methods utilize deep learning models to obtain code representations for vulnerability detection. However, these approaches predominantly capture the overall semantics of the code rather than its intrinsic vulnerabilit…

2025

MMLU-CF: A Contamination-free Multi-task Language Understanding Benchmark

ACL 2025long

Multiple-choice question (MCQ) datasets like Massive Multitask Language Understanding (MMLU) are widely used to evaluate the commonsense, understanding, and problem-solving abilities of large language models (LLMs). However, the open-source nature of these benchmarks and the broad sources of trainin…

2025

PEACE: Empowering Geologic Map Holistic Understanding with MLLMs

CVPR 2025poster

Geologic map, as a fundamental diagram in geology science, provides critical insights into the structure and composition of Earth's subsurface and surface. These maps are indispensable in various fields, including disaster assessment, resource exploration, and civil engineering. Despite their signif…

2025

Steering Large Language Models for Vulnerability Detection

ICASSP 2025accepted

Vulnerability detection remains a critical challenge in the field of security. Many existing approaches extract code representations for vulnerability detection. However, these methods often focus on the overall semantics of the code, neglecting to specifically target vulnerability-related semantics…

Cited by 0SourceScholar
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
2025

UCL-Bench: A Chinese User-Centric Legal Benchmark for Large Language Models

NAACL 2025findings

Existing legal benchmarks focusing on knowledge and logic effectively evaluate LLMs on various tasks in legal domain. However, few have explored the practical application of LLMs by actual users. To further assess whether LLMs meet the specific needs of legal practitioners in real-world scenarios, w…

2025

When and How to Augment Your Input: Question Routing Helps Balance the Accuracy and Efficiency of Large Language Models

NAACL 2025findings

Although large language models rely on parametric knowledge to achieve exceptional performance across various question-answering tasks, they still face challenges when addressing knowledge-based long-tail questions. Augmented generation techniques, such as chain-of-thought prompting and retrieval au…

Cited by 0SourcePDFScholar
2024

Mind's Eye of LLMs: Visualization-of-Thought Elicits Spatial Reasoning in Large Language Models

NeurIPS 2024poster

Large language models (LLMs) have exhibited impressive performance in language comprehension and various reasoning tasks. However, their abilities in spatial reasoning, a crucial aspect of human cognition, remain relatively unexplored. Human possess a remarkable ability to create mental images of un…

2024

TextDiffuser-2: Unleashing the Power of Language Models for Text Rendering

ECCV 2024oral

"The diffusion model has been proven a powerful generative model in recent years, yet it remains a challenge in generating visual text. Although existing work has endeavored to enhance the accuracy of text rendering, these methods still suffer from several drawbacks, such as (1) limited flexibility…

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

TextDiffuser: Diffusion Models as Text Painters

NeurIPS 2023poster

Diffusion models have gained increasing attention for their impressive generation abilities but currently struggle with rendering accurate and coherent text. To address this issue, we introduce TextDiffuser, focusing on generating images with visually appealing text that is coherent with backgrounds…

Cited by 128SourcePDFScholar
2023

TrOCR: Transformer-Based Optical Character Recognition with Pre-trained Models

AAAI 2023technical

Text recognition is a long-standing research problem for document digitalization. Existing approaches are usually built based on CNN for image understanding and RNN for char-level text generation. In addition, another language model is usually needed to improve the overall accuracy as a post-process…

2022

A Unified Model for Multi-class Anomaly Detection

NeurIPS 2022accept

Despite the rapid advance of unsupervised anomaly detection, existing methods require to train separate models for different objects. In this work, we present UniAD that accomplishes anomaly detection for multiple classes with a unified framework. Under such a challenging setting, popular reconstruc…

2022

Equal Loss: A Simple Loss Function for Noise Robust Learning

ICASSP 2022accepted

Training accurate deep neural networks in the presence of noisy labels is an important task. Though a number of approaches have been proposed for learning with noisy labels, many open issues remain. In this paper, we show that DNN learning with Cross Entropy is not robust to label noise and exhibits…

Cited by 0SourceScholar
2022

MarkupLM: Pre-training of Text and Markup Language for Visually Rich Document Understanding

ACL 2022long

Multimodal pre-training with text, layout, and image has made significant progress for Visually Rich Document Understanding (VRDU), especially the fixed-layout documents such as scanned document images. While, there are still a large number of digital documents where the layout information is not fi…

2022

XDoc: Unified Pre-training for Cross-Format Document Understanding

EMNLP 2022finding

The surge of pre-training has witnessed the rapid development of document understanding recently. Pre-training and fine-tuning framework has been effectively used to tackle texts in various formats, including plain texts, document texts, and web texts. Despite achieving promising performance, existi…

2022

XFUND: A Benchmark Dataset for Multilingual Visually Rich Form Understanding

ACL 2022findings

Multimodal pre-training with text, layout, and image has achieved SOTA performance for visually rich document understanding tasks recently, which demonstrates the great potential for joint learning across different modalities. However, the existed research work has focused only on the English domain…

2021

AutoSampling: Search for Effective Data Sampling Schedules

ICML 2021spotlight

Data sampling acts as a pivotal role in training deep learning models. However, an effective sampling schedule is difficult to learn due to its inherent high-dimension as a hyper-parameter. In this paper, we propose an AutoSampling method to automatically learn sampling schedules for model training,…

Cited by 8SourcePDFScholar
2021

Glancing at the Patch: Anomaly Localization With Global and Local Feature Comparison

CVPR 2021poster

Anomaly localization, with the purpose to segment the anomalous regions within images, is challenging due to the large variety of anomaly types. Existing methods typically train deep models by treating the entire image as a whole yet put little effort into learning the local distribution, which is v…

Cited by 85PDFScholar
2021

LayoutLMv2: Multi-modal Pre-training for Visually-rich Document Understanding

ACL 2021long

Pre-training of text and layout has proved effective in a variety of visually-rich document understanding tasks due to its effective model architecture and the advantage of large-scale unlabeled scanned/digital-born documents. We propose LayoutLMv2 architecture with new pre-training tasks to model t…

2021

LayoutReader: Pre-training of Text and Layout for Reading Order Detection

EMNLP 2021main

Reading order detection is the cornerstone to understanding visually-rich documents (e.g., receipts and forms). Unfortunately, no existing work took advantage of advanced deep learning models because it is too laborious to annotate a large enough dataset. We observe that the reading order of WORD do…

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…

2017

An underwater electrosensor for identifying objects of similar volume and aspect ratio using convolutional neural network

IROS 2017poster

Underwater electrosense is bio-inspired by weakly electric fishes that use an electric field to see the objects in the water. Current studies on engineering electrosense focus on designing sophisticated sensors and algorithms for emulating biological functions including localization and identificati…

Cited by 8SourceScholar