← Search

Siwei Wei

3 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
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…

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