← Search

Yan Cai

7 accepted papers

2026

DRPBench: Evaluating LLMs in Concurrent Code Comprehension via Fine-grained Data Race Prediction

ICML 2026spotlight

Large Language Models (LLMs) have demonstrated sophisticated comprehension of sequential code, yet their capacity for reasoning about concurrent programs remains largely unquantified. We introduce DRPBench, a benchmark designed to evaluate the concurrent code comprehension of LLMs by measuring their…

Cited by 0SourceScholar
2026

NextStep-1: Toward Autoregressive Image Generation with Continuous Tokens at Scale

ICLR 2026oral

Prevailing autoregressive (AR) models for text-to-image generation either rely on heavy, computationally-intensive diffusion models to process continuous image tokens, or employ vector quantization (VQ) to obtain discrete tokens with quantization loss. In this paper, we push the autoregressive parad…

Cited by 0SourcecodeScholar
2025

Training Deep Neural Networks with Virtual Smoothing Classes

AAAI 2025technical

Learning with softmax cross-entropy on one-hot labels often leads to overconfidence on the correct class. While label smoothing regulates this overconfidence by redistributing some confidence from the correct class to other incorrect classes, it compromises the representation in the logits about the…

2025

ViewSRD: 3D Visual Grounding via Structured Multi-View Decomposition

ICCV 2025poster

3D visual grounding aims to identify and localize objects in a 3D space based on textual descriptions. However, existing methods struggle with disentangling targets from anchors in complex multi-anchor queries and resolving inconsistencies in spatial descriptions caused by perspective variations.To…

2024

Extending Test-Time Augmentation with Metamorphic Relations for Combinatorial Problems

ICML 2024spotlight

The application of machine learning methods to solve combinatorial problems has garnered considerable research interest. In this paper, we propose MAgg (**M**etamorphic **Agg**regation), a method to augment machine learning models for combinatorial problems at inference time using metamorphic relati…

Cited by 0SourcePDFScholar
2024

MedBench: A Large-Scale Chinese Benchmark for Evaluating Medical Large Language Models

AAAI 2024technical

The emergence of various medical large language models (LLMs) in the medical domain has highlighted the need for unified evaluation standards, as manual evaluation of LLMs proves to be time-consuming and labor-intensive. To address this issue, we introduce MedBench, a comprehensive benchmark for the…

2023

What can Discriminator do? Towards Box-free Ownership Verification of Generative Adversarial Networks

ICCV 2023poster

In recent decades, Generative Adversarial Network (GAN) and its variants have achieved unprecedented success in image synthesis. However, well-trained GANs are under the threat of illegal steal or leakage. The prior studies on remote ownership verification assume a black-box setting where the defend…

Cited by 15PDFcodeScholar