← Search

Junyi An

4 accepted papers

2026

Periodic Bayesian Flow Networks with Additive Accuracy

ICML 2026poster

Generating periodic data---such as fractional atomic coordinates in crystal structures and phase patterns in compressive light-field (CLF) displays---is challenging because wrap-around boundaries complicate probabilistic modeling and learning. While Bayesian Flow Networks (BFNs) offer a powerful gen…

Cited by 0SourceScholar
2025

Equivariant Masked Position Prediction for Efficient Molecular Representation

ICLR 2025poster

Graph neural networks (GNNs) have shown considerable promise in computational chemistry. However, the limited availability of molecular data raises concerns regarding GNNs' ability to effectively capture the fundamental principles of physics and chemistry, which constrains their generalization capab…

2024

Hybrid Directional Graph Neural Network for Molecules

ICLR 2024spotlight

Equivariant message passing neural networks have emerged as the prevailing approach for predicting chemical properties of molecules due to their ability to leverage translation and rotation symmetries, resulting in a strong inductive bias. However, the equivariant operations in each layer can impose…