Post

Preprint! "Inverse Design of Complex Nanoparticle Heterostructures via Deep Learning on Heterogeneous Graphs" is now on ChemRxiv. In this work, we consider the problem of applying deep learning to heterogeneous nanostructures. (1/10) #ChemSky 🧪 #CompChem #physics #optics #nanoscience

doi.orgInverse Design of Complex Nanoparticle Heterostructures via Deep Learning on Heterogeneous GraphsApplications of deep learning (DL) to design nanomaterials are hampered by a lack of suitable data representations and training data. We report efforts to overcome these limitations and leverage DL to...

Towards this end, we performed thousands of high-fidelity kinetic Monte Carlo simulations (using RNMC, recently published in JOSS) and compiled them into a dataset called SUNSET - Simulated Upconverting Nanoparticle Spectra for Emission Tuning. (3/10)

With SUNSET, we can learn structure-property relationships for these complex nanostructures. We considered a number of different representations for nanoparticles - simple feature matrices, voxels (for convolutional neural networks), and graphs (for graph neural networks or GNNs). (4/10)

We also designed a new heterogeneous graph representation, with one set of nodes representing UCNP dopants and another set representing interactions within a layer or between layers. We found that our hetero-GNN outperformed other models in-distribution and especially out-of-distribution. (5/10)

While testing out these different models, we realized that UCNPs and similar core-shell particles display a unique "subdivision invariance". You can define layers arbitrarily, and the particle behavior doesn't change, but the representations (including for heterographs) do change. (6/10)

To address this, we included on-the-fly data augmentation, where the same particle would be included in a training batch with multiple layers divided in multiple different ways. This data augmentation allowed us to effectively learn subdivision invariance and lower our OOD error. (7/10)

0 reposts3 likes

With a trained hetero-GNN, we then got to the work of optimization. Because all of the structural features were included in our graph representation, we could directly perform gradient-based optimization, rapidly identifying particles far out of distribution with exceptional properties. (8/10)

These optimized particles included some with visible/UV emissions 6.5x higher than any UCNP in our training set! To ensure that these far-OOD predictions were reasonable, we validated by performing additional kMC simulations, finding generally good agreement. (9/10)

Congrats Eric, Lucas Attia (MIT ChemE), Benjamin Sanchez-Lengeling (now UToronto), Xiaojing Xia (postdoc at LBNL), Daniel Barter (now PsiQuantum), and of course Emory Chan (LBNL) and Sam - co-corresponding authors - for getting this out. Let's hope for swift and fair review in the new year. (10/10)