← Search

Bei Yu

60 accepted papers

2026

AnalogVerifier: A Neuro-Symbolic Framework for Analog Circuit Verification

ICML 2026poster

Analog circuits constitute the indispensable interface between physical reality and digital computation, underpinning safety-critical systems from autonomous driving to medical implants. Consequently, verification correctness is paramount; yet, it remains the critical bottleneck in hardware design, …

Cited by 0SourceScholar
2026

Beyond Majority Voting: Self-Reflective Test-Time Reinforcement Learning for LLM Reasoning

ICML 2026poster

The core challenge of Test-Time Reinforcement Learning (TTRL) lies in estimating rewards without access to ground-truth supervision. Existing TTRL methods predominantly rely on majority voting to generate pseudo-labels, under the assumption that the most frequent answer among sampled trajectories is…

Cited by 0SourceScholar
2026

CircuitNet 3.0: A Multi-Modal Dataset with Task-Oriented Augmentation for AI-Driven Circuit Design

ICLR 2026poster

Integrated circuit (IC) designs require transforming high-level specifications into physical layouts, demanding extensive expertise and specialized tools, as well as months of time and numerous iterations. While Machine Learning (ML) has shown promise in various research domains, the lack of large-s…

Cited by 0SourcecodeScholar
2026

DreamOmni2: Multimodal Instruction-based Generation and Editing

CVPR 2026

Recent advancements in instruction-based image editing and subject-driven generation have garnered significant attention, yet both tasks still face limitations in meeting practical user needs. Instruction-based editing relies solely on language instructions, which often fail to capture specific edit

Cited by 0SourcecodeScholar
2026

Expert-level Leaf Cell Layout Generation via Preference-Optimized LLM

ICML 2026poster

In the field of integrated circuits, leaf cells are the basic units, serving as the fundamental building blocks (e.g., standard cells) that are widely reused in various VLSI designs, forming the basis for more complex circuits. Therefore, the design quality of leaf cell layouts significantly impacts…

Cited by 0SourceScholar
2026

Fast Inverse Lithography via GRPO Reinforced Flow Matching

ICML 2026poster

In semiconductor manufacturing, lithography projects circuit layouts onto silicon wafers through an optical mask. As circuit features shrink below the wavelength of light, optical diffraction causes the printed patterns to deviate from their intended layouts. Inverse Lithography Technology (ILT) add…

Cited by 0SourceScholar
2026

KCLNet: Electrically Equivalence-Oriented Graph Representation Learning for Analog Circuits

AAAI 2026technical

Digital circuit representation learning has made remarkable progress in electronic design automation, effectively supporting critical tasks such as testability analysis and logic reasoning. However, representation learning for analog circuits remains challenging due to their continuous electrical ch

Cited by 0SourcePDFScholar
2026

MOSS: Efficient and Accurate FP8 LLM Training with Microscaling and Automatic Scaling

ICLR 2026poster

Training large language models with FP8 formats offers significant efficiency gains. However, the reduced numerical precision of FP8 poses challenges for stable and accurate training. Current frameworks preserve training performance using mixed-granularity quantization, i.e., applying per-group quan…

Cited by 0SourceScholar
2026

S-Quant: Rethinking Weight Quantization with Seed-Based Generation

ICML 2026poster

The progressive scaling of large language models (LLMs) has consistently enhanced multimodal understanding and advanced reasoning capabilities, but has substantially increased computational and hardware execution overhead. In this paper, we present S-Quant, a novel post-method that compresses only m…

Cited by 0SourceScholar
2026

SWINGARENA: Adversarial Programming Arena for Long-context GitHub Issue Solving

ICLR 2026oral

We present \textsc{SwingArena}, a adversarial evaluation framework for Large Language Models (LLMs) that closely mirrors real-world software development workflows. Unlike traditional static benchmarks, \textsc{SwingArena} models the collaborative process of software iteration by pairing LLMs as \tex…

Cited by 0SourcecodeScholar
2026

Stratified GRPO: Handling Structural Heterogeneity in Reinforcement Learning of LLM Search Agents

ICML 2026poster

Large language model (LLM) agents increasingly rely on external tools such as search engines to solve complex, multi-step problems, yet their rollouts are structurally heterogeneous: variations in tool-call number, placement, and outcomes induce distinct behaviors and reward distributions. As a resu…

Cited by 5SourceScholar
2026

Unlocking Token Rewards via Training-Free Reward Attribution

CVPR 2026

In this paper, we propose an extremely efficient, training-free method to extract token-level reward signals directly from an existing deep reward model. Our core idea is to attribute the overall process reward to individual tokens by estimating each token's influence. This influence is defined as t

Cited by 0SourcecodeScholar
2026

ViSurf: Visual Supervised-and-Reinforcement Fine-Tuning for Large Vision-and-Language Models

ICML 2026poster

Post-training Large Vision-and-Language Models (LVLMs) typically involves Supervised Fine-Tuning (SFT) for knowledge injection or Reinforcement Learning with Verifiable Rewards (RLVR) for performance enhancement. However, SFT often leads to sub-optimal performance, while RLVR remains constrained by …

Cited by 0SourceScholar
2026

VisionReasoner: Unified Reasoning-Integrated Visual Perception via Reinforcement Learning

ICLR 2026poster

Large vision-language models exhibit inherent capabilities to handle diverse visual perception tasks. In this paper, we introduce VisionReasoner, a unified framework capable of reasoning and solving multiple visual perception tasks within a shared model. Specifically, by designing a unified reward m…

Cited by 0SourcecodeScholar
2025

Bridging Layout and RTL: Knowledge Distillation based Timing Prediction

ICML 2025spotlight

Accurate and efficient timing prediction at the register-transfer level (RTL) remains a fundamental challenge in electronic design automation (EDA), particularly in striking a balance between accuracy and computational efficiency. While static timing analysis (STA) provides high-fidelity results thr…

2025

Circuit Representation Learning with Masked Gate Modeling and Verilog-AIG Alignment

ICLR 2025poster

Understanding the structure and function of circuits is crucial for electronic design automation (EDA). Circuits can be formulated as And-Inverter graphs (AIGs), enabling efficient implementation of representation learning through graph neural networks (GNNs). Masked modeling paradigms have been pr…

2025

DLoFT: Gradient-Decoupled Fine-Tuning for Generalizable Long Chain-of-Thought Reasoning

NeurIPS 2025poster

Long chain-of-thought (LongCoT) has emerged as a powerful reasoning paradigm for enabling large language models (LLMs) to solve complex tasks through a systematic and thorough thinking phase. Although supervised fine-tuning (SFT) on high-quality LongCoT traces has proven effective to activate LongCo…

Cited by 0SourceScholar
2025

Divergent Thoughts toward One Goal: LLM-based Multi-Agent Collaboration System for Electronic Design Automation

NAACL 2025long

Recently, with the development of tool-calling capabilities in large language models (LLMs), these models have demonstrated significant potential for automating electronic design automation (EDA) flows by interacting with EDA tool APIs via EDA scripts.However, considering the limited understanding o…

Cited by 1SourcePDFScholar
2025

Does Your Vision-Language Model Get Lost in the Long Video Sampling Dilemma?

ICCV 2025poster

The rise of Large Vision-Language Models (LVLMs) has significantly advanced video understanding. However, efficiently processing long videos remains a challenge due to the "Sampling Dilemma": low-density sampling risks missing critical information, while high-density sampling introduces redundancy.…

2025

DreamOmni: Unified Image Generation and Editing

CVPR 2025poster

Currently, the success of large language models (LLMs) illustrates that a unified multitasking approach can significantly enhance model usability, streamline deployment, and foster synergistic benefits across different tasks. However, in computer vision, while text-to-image (T2I) models have signifi…

Cited by 1SourcePDFScholar
2025

Efficient OpAmp Adaptation for Zoom Attention to Golden Contexts

ACL 2025long

Large language models (LLMs) have shown significant promise in question-answering (QA) tasks, particularly in retrieval-augmented generation (RAG) scenarios and long-context applications. However, their performance is hindered by noisy reference documents, which often distract from essential informa…

2025

Enhancing LLM Knowledge Learning through Generalization

EMNLP 2025

As Large language models (LLMs) are increasingly deployed in diverse applications, faithfully integrating evolving factual knowledge into these models remains a critical challenge. Continued pre-training on paraphrased data has shown empirical promise for enhancing knowledge acquisition. However, th

2025

Generative Video Propagation

CVPR 2025poster

Large-scale video generation models have the inherent ability to realistically model natural scenes. In this paper, we demonstrate that through a careful design of a generative video propagation framework, various video tasks can be addressed in a unified way by leveraging the generative power of su…

Cited by 1SourcePDFScholar
2025

Intelligent OPC Engineer Assistant for Semiconductor Manufacturing

AAAI 2025technical

Advancements in chip design and manufacturing have enabled the processing of complex tasks such as deep learning and natural language processing, paving the way for the development of artificial general intelligence (AGI). AI, on the other hand, can be leveraged to innovate and streamline semiconduc…

Cited by 1SourcePDFScholar
2025

Learnable Feature Patches and Vectors for Boosting Low-light Image Enhancement without External Knowledge

ICCV 2025poster

A major challenge in Low-Light Image Enhancement (LLIE) is its ill-posed nature: low-light images often lack sufficient information to align with normal-light ones (e.g., not all training data can be fully fitted to the ground truth). Numerous studies have attempted to bridge the gap between low- an…

Cited by 0SourcePDFScholar
2025

LithoSim: A Large, Holistic Lithography Simulation Benchmark for AI-Driven Semiconductor Manufacturing

NeurIPS 2025poster

Lithography orchestrates a symphony of light, mask and photochemicals to transfer the integrated circuit patterns onto the wafer. Lithography simulation serves as the critical nexus between circuit design and manufacturing, where its speed and accuracy fundamentally govern the optimization quality o…

Cited by 0SourcecodeScholar
2025

Low-Light Video Enhancement via Spatial-Temporal Consistent Decomposition

IJCAI 2025

Low-Light Video Enhancement (LLVE) seeks to restore dynamic or static scenes plagued by severe invisibility and noise. In this paper, we present an innovative video decomposition strategy that incorporates view-independent and view-dependent components to enhance the performance of LLVE. We leverage

Cited by 0SourcePDFScholar
2025

Mixture-of-Scores: Robust Image-Text Data Valuation via Three Lines of Code

ICCV 2025poster

Evaluating the quality of image-text pairs is essential for data processing in vision-language pre-training. Most metrics currently use off-the-shelf models, like CLIP-Score, to score pairs based on feature similarity. However, we find that different scoring models often produce inconsistent quality…

2025

On-Policy Optimization with Group Equivalent Preference for Multi-Programming Language Understanding

NeurIPS 2025poster

Large language models (LLMs) achieve remarkable performance in code generation tasks. However, a significant performance disparity persists between popular programming languages (e.g., Python, C++) and others. To address this capability gap, we leverage the code translation task to train LLMs, ther…

Cited by 0SourceScholar
2025

PermLLM: Learnable Channel Permutation for N:M Sparse Large Language Models

NeurIPS 2025poster

Channel permutation is a powerful technique for enhancing the accuracy of N:M sparse models by reordering the channels of weight matrices to prioritize the retention of important weights. However, traditional channel permutation methods rely on handcrafted quality metrics, which often fail to accur…

Cited by 0SourceScholar
2025

TGDPO: Harnessing Token-Level Reward Guidance for Enhancing Direct Preference Optimization

ICML 2025poster

Recent advancements in reinforcement learning from human feedback have shown that utilizing fine-grained token-level reward models can substantially enhance the performance of Proximal Policy Optimization (PPO) in aligning large language models. However, it is challenging to leverage such token-leve…

2025

VisionThink: Smart and Efficient Vision Language Model via Reinforcement Learning

NeurIPS 2025poster

Recent advancements in vision-language models (VLMs) have improved performance by increasing the number of visual tokens, which are often significantly longer than text tokens. However, we observe that most real-world scenarios do not require such an extensive number of visual tokens. While the perf…

Cited by 0SourceScholar
2025

VisionZip: Longer is Better but Not Necessary in Vision Language Models

CVPR 2025poster

Recent advancements in vision-language models have enhanced performance by increasing the length of visual tokens, making them much longer than text tokens and significantly raising computational costs.However, we observe that the visual tokens generated by popular vision encoders, such as CLIP and…

2024

BetterV: Controlled Verilog Generation with Discriminative Guidance

ICML 2024poster

Due to the growing complexity of modern Integrated Circuits (ICs), there is a need for automated circuit design methods. Recent years have seen increasing research in hardware design language generation to facilitate the design process. In this work, we propose a Verilog generation framework, Better…

Cited by 40SourcePDFScholar
2024

BiE: Bi-Exponent Block Floating-Point for Large Language Models Quantization

ICML 2024poster

Nowadays, Large Language Models (LLMs) mostly possess billions of parameters, bringing significant challenges to hardware platforms. Although quantization is an efficient approach to reduce computation and memory overhead for inference optimization, we stress the challenge that mainstream low-bit qu…

Cited by 5SourcePDFScholar
2024

Classes Are Not Equal: An Empirical Study on Image Recognition Fairness

CVPR 2024poster

In this paper we present an empirical study on image recognition unfairness i.e. extreme class accuracy disparity on balanced data like ImageNet. We demonstrate that classes are not equal and unfairness is prevalent for image classification models across various datasets network architectures and mo…

2024

Decoupled Kullback-Leibler Divergence Loss

NeurIPS 2024poster

In this paper, we delve deeper into the Kullback–Leibler (KL) Divergence loss and mathematically prove that it is equivalent to the Decoupled Kullback-Leibler (DKL) Divergence loss that consists of 1) a weighted Mean Square Error ($\mathbf{w}$MSE) loss and 2) a Cross-Entropy loss incorporating soft…

2024

Improving Neural ODE Training with Temporal Adaptive Batch Normalization

NeurIPS 2024poster

Neural ordinary differential equations (Neural ODEs) is a family of continuous-depth neural networks where the evolution of hidden states is governed by learnable temporal derivatives. We identify a significant limitation in applying traditional Batch Normalization (BN) to Neural ODEs, due to a fund…

Cited by 1SourcePDFScholar
2024

Parameter-Efficient Sparsity Crafting from Dense to Mixture-of-Experts for Instruction Tuning on General Tasks

EMNLP 2024main

Large language models (LLMs) have demonstrated considerable proficiency in general natural language processing (NLP) tasks. Instruction tuning, a successful paradigm, enhances the ability of LLMs to follow natural language instructions and exhibit robust generalization across general tasks. However,…

2024

Progressively Knowledge Distillation via Re-parameterizing Diffusion Reverse Process

AAAI 2024technical

Knowledge distillation aims at transferring knowledge from the teacher model to the student one by aligning their distributions. Feature-level distillation often uses L2 distance or its variants as the loss function, based on the assumption that outputs follow normal distributions. This poses a si…

Cited by 1SourcePDFScholar
2024

Towards Automated RISC-V Microarchitecture Design with Reinforcement Learning

AAAI 2024technical

Microarchitecture determines the implementation of a microprocessor. Designing a microarchitecture to achieve better performance, power, and area (PPA) trade-off has been increasingly difficult. Previous data-driven methodologies hold inappropriate assumptions and lack more tightly coupling with exp…

2024

p-Laplacian Adaptation for Generative Pre-trained Vision-Language Models

AAAI 2024technical

Vision-Language models (VLMs) pre-trained on large corpora have demonstrated notable success across a range of downstream tasks. In light of the rapidly increasing size of pre-trained VLMs, parameter-efficient transfer learning (PETL) has garnered attention as a viable alternative to full fine-tunin…

2023

ATFormer: A Learned Performance Model with Transfer Learning Across Devices for Deep Learning Tensor Programs

EMNLP 2023long main

The training and inference efficiency of ever-larger deep neural networks highly rely on the performance of tensor operators on specific hardware platforms. Therefore, a compilation-based optimization flow with automatic tensor generation and parameter tuning is necessary for efficient model deploym…

Cited by 0SourceScholar
2023

AutoGraph: Optimizing DNN Computation Graph for Parallel GPU Kernel Execution

AAAI 2023technical

Deep learning frameworks optimize the computation graphs and intra-operator computations to boost the inference performance on GPUs, while inter-operator parallelism is usually ignored. In this paper, a unified framework, AutoGraph, is proposed to obtain highly optimized computation graphs in favo…

Cited by 6SourcePDFScholar
2023

Do Not Train It: A Linear Neural Architecture Search of Graph Neural Networks

ICML 2023poster

Neural architecture search (NAS) for Graph neural networks (GNNs), called NAS-GNNs, has achieved significant performance over manually designed GNN architectures. However, these methods inherit issues from the conventional NAS methods, such as high computational cost and optimization difficulty. Mor…

2023

FastGR: Global Routing on CPU-GPU with Heterogeneous Task Graph Scheduler (Extended Abstract)

IJCAI 2023poster

Running time is a key metric across the standard physical design flow stages. However, with the rapid growth in design sizes, routing runtime has become the runtime bottleneck in the physical design flow. To improve the effectiveness of the modern global router, we propose a global routing framework…

Cited by 0SourcePDFScholar
2023

LithoBench: Benchmarking AI Computational Lithography for Semiconductor Manufacturing

NeurIPS 2023poster

Computational lithography provides algorithmic and mathematical support for resolution enhancement in optical lithography, which is the critical step in semiconductor manufacturing. The time-consuming lithography simulation and mask optimization processes limit the practical application of inverse…

2022

Context-Based Contrastive Learning for Scene Text Recognition

AAAI 2022technical

Pursuing accurate and robust recognizers has been a long-lasting goal for scene text recognition (STR) researchers. Recently, attention-based methods have demonstrated their effectiveness and achieved impressive results on public benchmarks. The attention mechanism enables models to recognize scene…

Cited by 62SourcePDFScholar
2022

PCL: Proxy-Based Contrastive Learning for Domain Generalization

CVPR 2022poster

Domain generalization refers to the problem of training a model from a collection of different source domains that can directly generalize to the unseen target domains. A promising solution is contrastive learning, which attempts to learn domain-invariant representations by exploiting rich semantic…

Cited by 157PDFcodeScholar
2021

Fast and Efficient DNN Deployment via Deep Gaussian Transfer Learning

ICCV 2021poster

Deep neural networks (DNNs) have been widely used recently while their hardware deployment optimizations are very time-consuming and the historical deployment knowledge is not utilized efficiently. In this paper, to accelerate the optimization process and find better deployment configurations, we pr…

Cited by 7PDFScholar
2021

Seeing Dynamic Scene in the Dark: A High-Quality Video Dataset With Mechatronic Alignment

ICCV 2021poster

Low-light video enhancement is an important task. Previous work is mostly trained on paired static images or videos. We compile a new dataset formed by our new strategy that contains high-quality spatially-aligned video pairs from dynamic scenes in low- and normal-light conditions. We built it using…

Cited by 117PDFcodeScholar
2020

Tensor Low-Rank Reconstruction for Semantic Segmentation

ECCV 2020poster

Context information plays an indispensable role in the success of semantic segmentation. Recently, non-local self-attention based methods are proved to be effective for context information collection. Since desired context consists of spatial-wise and channel-wise attentions, the 3D representation i…

Cited by 88SourcePDFScholar