← Search

Yanjun Wu

20 accepted papers

2026

Behavioral Embeddings of Programs: A Quasi-Dynamic Approach for Optimization Prediction

ICLR 2026poster

Learning effective numerical representations, or embeddings, of programs is a fundamental prerequisite for applying machine learning to automate and enhance compiler optimization. Prevailing paradigms, however, present a dilemma. Static representations, derived from source code or intermediate repre…

Cited by 0SourcecodeScholar
2026

ECCO: Evidence-Driven Causal Reasoning for Compiler Optimization

ICML 2026poster

Compiler auto-tuning faces a dichotomy between traditional black-box search methods, which lack semantic guidance, and recent Large Language Model (LLM) approaches, which often suffer from superficial pattern matching and causal opacity. In this paper, we introduce ECCO, a framework that bridges int…

Cited by 0SourceScholar
2026

HLD: Approximate Hierarchical Linguistic Distribution Modeling for LLM-Generated Text Detection

ICLR 2026poster

The widespread deployment of large language models (LLMs) has made the reliable detection of AI-generated text a crucial task. However, existing zero-shot detectors typically rely on proxy models to approximate probability distributions of unknown source models at a single token level. Such approach…

Cited by 0SourcecodeScholar
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

Compiler-R1: Towards Agentic Compiler Auto-tuning with Reinforcement Learning

NeurIPS 2025poster

Compiler auto-tuning optimizes pass sequences to improve performance metrics such as Intermediate Representation (IR) instruction count. Although recent advances leveraging Large Language Models (LLMs) have shown promise in automating compiler tuning, two significant challenges still remain: the abs…

Cited by 0SourcecodeScholar
2025

EasySpec: Layer-Parallel Speculative Decoding for Efficient Multi-GPU Utilization

NeurIPS 2025poster

Speculative decoding is an effective and lossless method for Large Language Model (LLM) inference acceleration. It employs a smaller model to generate a draft token sequence, which is then verified by the original base model. In multi-GPU systems, inference latency can be further reduced through ten…

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

SEEN-DA: SEmantic ENtropy guided Domain-aware Attention for Domain Adaptive Object Detection

CVPR 2025poster

Domain adaptive object detection (DAOD) aims to generalize detectors trained on an annotated source domain to an unlabelled target domain. Traditional works focus on aligning visual features between domains to extract domain-invariant knowledge, and recent VLM-based DAOD methods leverage semantic in…

Cited by 0SourcePDFScholar
2024

BPDO: Boundary Points Dynamic Optimization for Arbitrary Shape Scene Text Detection

ICASSP 2024accepted

Arbitrary shape scene text detection is of great importance in scene understanding tasks. Due to the complexity and diversity of text in natural scenes, existing scene text algorithms have limited accuracy for detecting arbitrary shape text. In this paper, we propose a novel arbitrary shape scene te…

Cited by 0SourceScholar
2024

Text Region Multiple Information Perception Network for Scene Text Detection

ICASSP 2024accepted

Segmentation-based scene text detection algorithms can handle arbitrary shape scene texts and have strong robustness and adaptability, so it has attracted wide attention. Existing segmentation-based scene text detection algorithms usually only segment the pixels in the center region of the text, whi…

Cited by 0SourceScholar
2022

LD-ConGR: A Large RGB-D Video Dataset for Long-Distance Continuous Gesture Recognition

CVPR 2022poster

Gesture recognition plays an important role in natural human-computer interaction and sign language recognition. Existing research on gesture recognition is limited to close-range interaction such as vehicle gesture control and face-to-face communication. To apply gesture recognition to long-distanc…

Cited by 23PDFcodeScholar
2022

Multi-Granularity Alignment Domain Adaptation for Object Detection

CVPR 2022poster

Domain adaptive object detection is challenging due to distinctive data distribution between source domain and target domain. In this paper, we propose a unified multi-granularity alignment based object detection framework towards domain-invariant feature learning. To this end, we encode the depende…

Cited by 101PDFcodeScholar
2021

Towards Real-World Prohibited Item Detection: A Large-Scale X-Ray Benchmark

ICCV 2021poster

Automatic security inspection using computer vision technology is a challenging task in real-world scenarios due to various factors, including intra-class variance, class imbalance, and occlusion. Most of the previous methods rarely solve the cases that the prohibited items are deliberately hidden i…

Cited by 116PDFcodeScholar
2020

Attention Convolutional Binary Neural Tree for Fine-Grained Visual Categorization

CVPR 2020poster

Fine-grained visual categorization (FGVC) is an important but challenging task due to high intra-class variances and low inter-class variances caused by deformation, occlusion, illumination, etc. An attention convolutional binary neural tree architecture is presented to address those problems for we…

Cited by 276PDFScholar
2020

Learning Semantic Neural Tree for Human Parsing

ECCV 2020poster

In this paper, we design a novel semantic neural tree for human parsing, which uses a tree architecture to encode physiological structure of human body, and design a coarse to fine process in a cascade manner to generate accurate results. Specifically, the semantic neural tree is designed to segment…

Cited by 71SourcePDFScholar
2020

Spatial Attention Pyramid Network for Unsupervised Domain Adaptation

ECCV 2020poster

Unsupervised domain adaptation is critical in various computer vision tasks, such as object detection, instance segmentation, and semantic segmentation, which aims to alleviate performance degradation caused by domain-shift. Most of previous methods rely on a single-mode distribution of source and t…

Cited by 137SourcePDFScholar