Saturday, December 11, 2021

Real progress on machine learning for density functional theory

(Sorry about the slow pace of posting.  The end of the semester has been very intense, including a faculty retreat for our department last week.)

I've written before (here, here, and here) about density functional theory, arguably one of the most impactful intellectual physics results of 20th century physics.   DFT is one approach to trying to solve the quantum electronic structure problem for molecules or solids containing many electrons.  As explained in the links above, the idea is powerful.  It turns out that the ground state (lowest energy state) electronic density as a function of position \(n(\mathbf{r})\), contains all the information needed to calculate basically anything you could want to know about the ground state.  There is a functional \(E[n(\mathbf{r})]\), for example, that will give you the energy of the full-on, interacting many-electron ground state.  It's possible to do a non-interacting electron model that can get you arbitrarily close to the true, correct \(n(\mathbf{r})\), The tricky bit is, there is no exact analytical expression for the functional \(E[n(\mathbf{r})]\), which includes a particularly tricky contribution called the exchange-correlation part of the functional, \(E_{\mathrm{xc}}[n(\mathbf{r})]\).  Because we are talking about functionals rather than functions,  \(E_{\mathrm{xc}}[n(\mathbf{r})]\) might depend in a non-local way on \(n(\mathbf{r})\) and its derivatives at all points in space - there is no reason to think it will be simple to write down.  

From deepmind.com

I wrote six years ago about the idea that machine learning techniques might make it possible to get a working version of something close to the exact \(E_{\mathrm{xc}}[n(\mathbf{r})]\) , even if we can't readily write it down in some closed form.  Now it seems that real progress has been made in this direction.  Here is a blog post from the DeepMind team about their paper in Science this week where they demonstrate a new functional that they claim is very good and accurate vs exact calculations on test systems, computationally tractable, and satisfies fundamental properties that have to hold for the true exact functional.  They argue that their code is more than just a fancy look-up table and that it contains generalizable knowledge so that it's useful well beyond their specific training test cases.  

If this is so, then it could be a major step forward in (for some definitions of the term) first-principles calculations of molecular and material properties.  I'm curious about whether the new functional will actually let us gain some physical insight into why physics requires that particular underlying mathematical structure.  Still, even if we end up with a "black box" that allows greatly improved calculations, that would really be something.  I'd appreciate it if knowledgable DFT/electronic structure experts could comment here on how excited we should be about this.



5 comments:

Pizza Perusing Physicist said...
This comment has been removed by the author.
Michael Swift said...

The perspective by Perdew is a good read.

I would say this is a pretty big deal, but mostly as a proof-of-concept. The general approach is:
1. Identify a shortcoming of all known functionals (Figure 2: failure to describe bond-breaking)
2. Identify the source (existing functionals fail to simultaneously satisfy certain mathematical constraints)
3. Train a machine-learned functional that satisfies the constraints, using data from a method without the shortcoming (coupled-cluster methods)
4. Show that the new functional corrects the shortcoming
5. Show that the new functional can generalize beyond the training data (Figure 3: DNA base pairs, compressed hydrogen chain, and a class of reaction barriers)

This paper is exciting because (as far as I know) this is the first successful demonstration of this approach. Unfortunately DM21 itself is not going to revolutionize chemistry or materials science, because the examples in (5) can also be solved by the beyond-DFT methods in (3). The dream is for a similar approach to result in a functional that can generalize to problems beyond the reach of the method used to generate the training data. This is still likely a ways away, but the success of DM21 makes it seem a lot more plausible.

An interesting question is computational cost. The authors don't talk about this, but my guess is that there will be a modest cost savings compared to CCSD(T). Coupled-cluster is very expensive and neural networks (once trained) are very cheap, but DM21 depends on Hartree-Fock features, which I expect are somewhat expensive to calculate.

Another important note: Perdew points out that DM21 won't work for transition-metal chemistry, for solids, or for liquids, because those problems depend on physics that wasn't in the training data.

Michael Swift said...

Strongly correlated solid-state systems are a place where this approach could be really valuable. Existing functionals fail and we know (at least partially) the broken constraints behind the failure. We have some methods that work (e.g. DMRG, DMFT) but they are very limited in scope. If a machine-learned functional could be trained to describe strongly correlated systems that are beyond our current modeling capability, that would be exciting indeed!

Anonymous said...

Hi Doug, I like your blog posts, but with all due respect, can you please stop apologizing for the "slow pace of posting"? Too many posts start this way, and it's really unnecessary.


Regina zalit said...

I never saw myself doing this months ago, but because I'm happy and want others to find their peace of mind in their matrimonial homes or relationship I decided to share my encounter with Dr Osita . I've been married with three kids for 13 years and things were really great with my husband. About 3 years ago my husband started lying, cheating, womanizing outside our marriage and I wasn't comfortable with it, so I stayed one night thinking sleeplessly. I took my phone and was going online when I saw a write up which was titled. PROBLEMS THAT MOST PEOPLE COMMONLY HAVE IN MARRIAGE BEEN SOLVED BY DOCTOR Osita. I contacted him and discussed my marriage issues and he concluded that a spell was cast on my husband by another woman" whose husband is no more, so we went through the processes and some incantation was done to free my husband and this made my husband stopped his misconduct in our marriage and now my husband has been on his real self, this testimony is to advise people out there to never break up with their marriage but to look for solutions in the right places. you can contact Dr Osita on Whatsapp / Viber :+4915217824272 Or Email (Ositabello8@gmil.com)