← Search

Ba-Hien Tran

7 accepted papers

2026

Optimizing Data Augmentation through Bayesian Model Selection

ICLR 2026poster

Data Augmentation (DA) has become an essential tool to improve robustness and generalization of modern machine learning. However, when deciding on DA strategies it is critical to choose parameters carefully, and this can be a daunting task which is traditionally left to trial-and-error or expensive…

Cited by 0SourceScholar
2025

Robust Classification by Coupling Data Mollification with Label Smoothing

AISTATS 2025poster

Introducing training-time augmentations is a key technique to enhance generalization and prepare deep neural networks against test-time corruptions. Inspired by the success of generative diffusion models, we propose a novel approach of coupling data mollification, in the form of image noising and bl…

Cited by 0SourcecodeScholar
2023

Fully Bayesian Autoencoders with Latent Sparse Gaussian Processes

ICML 2023poster

We present a fully Bayesian autoencoder model that treats both local latent variables and global decoder parameters in a Bayesian fashion. This approach allows for flexible priors and posterior approximations while keeping the inference costs low. To achieve this, we introduce an amortized MCMC appr…

Cited by 7SourcePDFScholar
2023

One-Line-of-Code Data Mollification Improves Optimization of Likelihood-based Generative Models

NeurIPS 2023poster

Generative Models (GMs) have attracted considerable attention due to their tremendous success in various domains, such as computer vision where they are capable to generate impressive realistic-looking images. Likelihood-based GMs are attractive due to the possibility to generate new data by a singl…

2021

Model Selection for Bayesian Autoencoders

NeurIPS 2021poster

We develop a novel method for carrying out model selection for Bayesian autoencoders (BAEs) by means of prior hyper-parameter optimization. Inspired by the common practice of type-II maximum likelihood optimization and its equivalence to Kullback-Leibler divergence minimization, we propose to optimi…