Inverse molecular design and parameter optimization with Hückel theory using automatic differentiation

Rodrigo A. Vargas-Hernández*, Kjell Jorner, Robert Pollice, Alán Aspuru-Guzik

*Corresponding author for this work

    Research output: Contribution to journalArticleAcademicpeer-review

    4 Citations (Scopus)
    58 Downloads (Pure)


    Semiempirical quantum chemistry has recently seen a renaissance with applications in high-throughput virtual screening and machine learning. The simplest semiempirical model still in widespread use in chemistry is Hückel's π-electron molecular orbital theory. In this work, we implemented a Hückel program using differentiable programming with the JAX framework based on limited modifications of a pre-existing NumPy version. The auto-differentiable Hückel code enabled efficient gradient-based optimization of model parameters tuned for excitation energies and molecular polarizabilities, respectively, based on as few as 100 data points from density functional theory simulations. In particular, the facile computation of the polarizability, a second-order derivative, via auto-differentiation shows the potential of differentiable programming to bypass the need for numeric differentiation or derivation of analytical expressions. Finally, we employ gradient-based optimization of atom identity for inverse design of organic electronic materials with targeted orbital energy gaps and polarizabilities. Optimized structures are obtained after as little as 15 iterations using standard gradient-based optimization algorithms.

    Original languageEnglish
    Article number104801
    Number of pages17
    JournalJournal of Chemical Physics
    Issue number10
    Publication statusPublished - 14-Mar-2023

    Cite this