Polyatomic Complexes: A topologically-informed learning representation for atomistic systems

Khorana, Rahul, Noack, Marcus, Qian, Jin

arXiv.org Artificial Intelligence 

Developing robust representations of chemical structures that enable models to learn topological inductive biases is challenging. In this manuscript, we present a representation of atomistic systems. We begin by proving that our representation satisfies all structural, geometric, efficiency, and generalizability constraints. Afterward, we provide a general algorithm to encode any atomistic system. Finally, we report performance comparable to state-of-the-art methods on numerous tasks.