← Search

Haoyue Bai

15 accepted papers

2026

Brownian Bridge Augmented Surrogate Simulation and Injection Planning for Geological CO2 Storage

AAAI 2026technical

Geological CO2 storage (GCS) involves injecting captured CO2 into deep subsurface formations to support climate goals. The effective management of GCS relies on adaptive injection planning to dynamically control injection rates and well pressures to balance both storage safety and efficiency. Prior

Cited by 0SourcePDFScholar
2026

DELTA-Code: How RL Unlocks and Transfers New Programming Algorithms in LLMs

ICLR 2026poster

It remains an open question whether LLMs can acquire or generalize genuinely new reasoning strategies, beyond the sharpened skills encoded in their parameters during pre-training or post-training. To attempt to answer this debate, we introduce DELTA-Code —Distributional Evaluation of Learnability an…

Cited by 0SourcecodeScholar
2026

Efficient Post-Training Refinement of Latent Reasoning in Large Language Models

AAAI 2026technical

Reasoning is a key component of language understanding in Large Language Models. While Chain-of-Thought prompting enhances performance via explicit intermediate steps, it suffers from sufficient token overhead and a fixed reasoning trajectory, preventing step-wise refinement. Recent advances in late

Cited by 0SourcePDFScholar
2026

Lightweight Guidance Sampling and Deep Refinement Reconstruction Network for Adaptive Compressive Sensing

ICRA 2026poster

Adaptive Compressive Sensing (ACS) has attracted increasing attention for its ability to progressively improve image reconstruction quality by dynamically adjusting sampling allocation. Multi-stage sampling is a promising strategy that leverages intermediate reconstructions to guide sampling without…

Cited by 0Scholar
2025

Sculpting Features from Noise: Reward-Guided Hierarchical Diffusion for Task-Optimal Feature Transformation

NeurIPS 2025poster

Feature Transformation (FT) crafts new features from original ones via mathematical operations to enhance dataset expressiveness for downstream models. However, existing FT methods exhibit critical limitations: discrete search struggles with enormous combinatorial spaces, impeding practical use; and…

Cited by 0SourcecodeScholar
2025

Unsupervised Feature Transformation via In-context Generation, Generator-critic LLM Agents, and Duet-play Teaming

IJCAI 2025

Feature transformation involves generating a new set of features from the original dataset to enhance the data's utility. In certain domains like material performance screening, dimensionality is large and collecting labels is expensive and lengthy. It highly necessitates transforming feature spaces

2025

Where's the Liability in the Generative Era? Recovery-based Black-Box Detection of AI-Generated Content

CVPR 2025poster

The recent proliferation of photorealistic images created by generative models has sparked both excitement and concern, as these images are increasingly indistinguishable from real ones to the human eye. While offering new creative and commercial possibilities, the potential for misuse, such as in m…

Cited by 0SourcePDFScholar
2024

HYPO: Hyperspherical Out-Of-Distribution Generalization

ICLR 2024poster

Out-of-distribution (OOD) generalization is critical for machine learning models deployed in the real world. However, achieving this can be fundamentally challenging, as it requires the ability to learn invariant features across different domains or environments. In this paper, we propose a novel fr…

2023

Feed Two Birds with One Scone: Exploiting Wild Data for Both Out-of-Distribution Generalization and Detection

ICML 2023poster

Modern machine learning models deployed in the wild can encounter both covariate and semantic shifts, giving rise to the problems of out-of-distribution (OOD) generalization and OOD detection respectively. While both problems have received significant research attention lately, they have been pursue…

2022

OoD-Bench: Quantifying and Understanding Two Dimensions of Out-of-Distribution Generalization

CVPR 2022oral

Deep learning has achieved tremendous success with independent and identically distributed (i.i.d.) data. However, the performance of neural networks often degenerates drastically when encountering out-of-distribution (OoD) data, i.e., when training and test data are sampled from different distribut…

Cited by 125PDFcodeScholar
2021

DecAug: Out-of-Distribution Generalization via Decomposed Feature Representation and Semantic Augmentation

AAAI 2021technical

While deep learning demonstrates its strong ability to handle independent and identically distributed (IID) data, it often suffers from out-of-distribution (OoD) generalization, where the test data come from another distribution (w.r.t. the training one). Designing a general OoD generalization frame…

Cited by 86SourcePDFScholar
2021

NAS-OoD: Neural Architecture Search for Out-of-Distribution Generalization

ICCV 2021poster

Recent advances on Out-of-Distribution (OoD) generalization reveal the robustness of deep learning models against distribution shifts. However, existing works focus on OoD algorithms, such as invariant risk minimization, domain generalization, or stable learning, without considering the influence of…

Cited by 56PDFScholar
2021

Pyramid R-CNN: Towards Better Performance and Adaptability for 3D Object Detection

ICCV 2021poster

We present a flexible and high-performance framework, named Pyramid R-CNN, for two-stage 3D object detection from point clouds. Current approaches generally rely on the points or voxels of interest for RoI feature extraction on the second stage, but cannot effectively handle the sparsity and non-uni…

Cited by 201PDFcodeScholar