← Search

Yuanbo Wen

19 accepted papers

2026

PerceptOS: Semantic-Aware Kernel Optimization for OS-Intensive Workloads via Hardware-Software Alignment

ICML 2026poster

Optimizing OS kernels for specific applications is vital for peak performance, yet existing LLM-based methods struggle with a semantic mismatch between generalized reasoning and low-level system behaviors. As a result, these static, open-loop approaches suffer from runtime blindness, configuration f…

Cited by 0SourceScholar
2026

QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generation

AAAI 2026technical

Developing high-performance GPU kernels is critical for AI and scientific computing, but remains challenging due to its reliance on expert crafting and poor portability. While large language models (LLMs) offer promise for automation, both general-purpose and finetuned LLMs suffer from two fundament

Cited by 0SourcePDFScholar
2025

Automated Superscalar Processor Design by Learning Data Dependencies

IJCAI 2025

Automated processor design, which can significantly reduce human efforts and accelerate design cycles, has received considerable attention. While recent advancements have automatically designed single-cycle processors that execute one instruction per cycle, their performance cannot compete with mode

Cited by 0SourcePDFScholar
2025

Brain-Inspired Spiking Neural Networks for Energy-Efficient Object Detection

CVPR 2025poster

Brain-inspired spiking neural networks (SNNs) have the capability of energy-efficient processing of temporal information. However, leveraging the rich dynamic characteristics of SNNs and prior works in artificial neural networks (ANNs) to construct an effective object detection model for visual task…

Cited by 0SourcePDFScholar
2025

MigGPT: Harnessing Large Language Models for Automated Migration of Out-of-Tree Linux Kernel Patches Across Versions

NeurIPS 2025spotlight

Out-of-tree kernel patches are essential for adapting the Linux kernel to new hardware or enabling specific functionalities. Maintaining and updating these patches across different kernel versions demands significant effort from experienced engineers. Large language models (LLMs) have shown remarkab…

Cited by 0SourceScholar
2025

Multi-axis Prompt and Multi-dimension Fusion Network for All-in-one Weather-degraded Image Restoration

AAAI 2025technical

Existing approaches aiming to remove adverse weather degradations compromise the image quality and incur the long processing time. To this end, we introduce a multi-axis prompt and multi-dimension fusion network (MPMF-Net). Specifically, we develop a multi-axis prompts learning block (MPLB), which l…

2025

QiMeng-Attention: SOTA Attention Operator is generated by SOTA Attention Algorithm

ACL 2025finding

The attention operator remains a critical performance bottleneck in large language models (LLMs), particularly for long-context scenarios. While FlashAttention is the most widely used and effective GPU-aware acceleration algorithm, it must require time-consuming and hardware-specific manual implemen…

2025

QiMeng-GEMM: Automatically Generating High-Performance Matrix Multiplication Code by Exploiting Large Language Models

AAAI 2025technical

As a crucial operator in numerous scientific and engineering computing applications, the automatic optimization of General Matrix Multiplication (GEMM) with full utilization of ever-evolving hardware architectures (e.g. GPUs and RISC-V) is of paramount importance. While Large Language Models (LLMs)…

Cited by 0SourcePDFScholar
2025

QiMeng-MuPa: Mutual-Supervised Learning for Sequential-to-Parallel Code Translation

NeurIPS 2025poster

The rise of GPU-based high-performance computing (HPC) has driven the widespread adoption of parallel programming models such as CUDA. Yet, the inherent complexity of parallel programming creates a demand for the automated sequential-to-parallel approaches. However, data scarcity poses a significant…

Cited by 0SourcecodeScholar
2025

QiMeng-NeuComBack: Self-Evolving Translation from IR to Assembly Code

NeurIPS 2025poster

Compilers, while essential, are notoriously complex systems that demand prohibitively expensive human expertise to develop and maintain. The recent advancements in Large Language Models (LLMs) offer a compelling new paradigm: Neural Compilation, which could potentially simplify compiler development…

Cited by 0SourceScholar
2025

QiMeng-TensorOp: One-Line Prompt is Enough for High-Performance Tensor Operator Generation with Hardware Primitives

IJCAI 2025

Computation-intensive tensor operators constitute over 90% of the computations in Large Language Models (LLMs) and Deep Neural Networks. Automatically and efficiently generating high-performance tensor operators with hardware primitives is crucial for diverse and ever-evolving hardware architectures

Cited by 0SourcePDFScholar
2024

AutoOS: Make Your OS More Powerful by Exploiting Large Language Models

ICML 2024poster

With the rapid development of Artificial Intelligence of Things (AIoT), customizing and optimizing operating system (OS) kernel configurations for various AIoT application scenarios is crucial for maximizing system performance. However, existing approaches falter due to the overwhelming problem comp…

Cited by 3SourcePDFScholar
2024

Encoder-Minimal and Decoder-Minimal Framework for Remote Sensing Image Dehazing

ICASSP 2024accepted

Haze obscures remote sensing images, hindering valuable information extraction. To this end, we propose RSHazeNet, an encoder-minimal and decoder-minimal framework for efficient remote sensing image dehazing. Specifically, regarding the process of merging features within the same level, we develop a…

Cited by 0SourceScholar
2024

Multi-Dimension Queried and Interacting Network for Stereo Image Deraining

ICASSP 2024accepted

Eliminating the rain degradation in stereo images poses a formidable challenge, which necessitates the efficient exploitation of mutual information present between the dual views. To this end, we devise MQINet, which employs multi-dimension queries and interactions for stereo image deraining. More s…

Cited by 0SourceScholar
2024

Prompt-based Visual Alignment for Zero-shot Policy Transfer

ICML 2024poster

Overfitting in RL has become one of the main obstacles to applications in reinforcement learning(RL). Existing methods do not provide explicit semantic constrain for the feature extractor, hindering the agent from learning a unified cross-domain representation and resulting in performance degradatio…

Cited by 0SourcePDFScholar
2023

ANPL: Towards Natural Programming with Interactive Decomposition

NeurIPS 2023poster

Though LLMs are capable of generating plausible programs, it’s challenging to interact with the LLMs further to revise the program, especially if the user’s specific requirements are different from the initial proposal. In this paper, we introduce ANPL, an interactive programming system that ensures…

2023

BALTO: fast tensor program optimization with diversity-based active learning

ICLR 2023poster

Tensor program optimization (TPO) based on pre-trained models can effectively reduce the computing time of deep neural networks. However, training of such models is prohibitively expensive, which highly depends on a large-scale dataset and thus requires tremendous time-consuming performance measurem…

Cited by 6SourcePDFScholar
2023

Emergent Communication for Rules Reasoning

NeurIPS 2023poster

Research on emergent communication between deep-learning-based agents has received extensive attention due to its inspiration for linguistics and artificial intelligence. However, previous attempts have hovered around emerging communication under perception-oriented environmental settings, that…

Cited by 0SourcePDFScholar
2022

BabelTower: Learning to Auto-parallelized Program Translation

ICML 2022spotlight

GPUs have become the dominant computing platforms for many applications, while programming GPUs with the widely-used CUDA parallel programming model is difficult. As sequential C code is relatively easy to obtain either from legacy repositories or by manual implementation, automatically translating…

Cited by 13SourcePDFScholar