What is the role of Bayesian ML for AI alignment/safety?

I began my PhD with a focus on Bayesian deep learning with exactly the same reasoning as you. I also share your doubts about the relevance of BDL to long-term safety. I have two clusters of thoughts: some reasons why BDL might be worth pursuing regardless, and alternative approaches.

Considerations about BDL and important safety research:

• Don't overfit to recent trends. LLMs are very remarkable. Before them, DRL was very remarkable. I don't know what will be remarkable next. My hunch is that we won't get AGI by just doing more of what we are doing now. (People I respect disagree with that, and I am uncertain. Also, note I don't say we could't get AGI that way.)
• Bayesian inference is powerful and general. The original motivation is still real. It is tempered by your (in my view, correct) observation that existing methods for approximate inference have big flaws. My view is that probability still describes the correct way to update given evidence and so it contains deep truths about reliable information processing. That means that understanding approximate Bayesian inference is still a useful guide for anyone trying to automatically process information correctly (and being aware of the necessary assumptions). And an awful lot of failure modes for AGI involve dangerous mistaken generalization. Also note that statements like "simple non-Bayesian techniques such as ensembles" are controversial, and there's considerable debate about whether ensembles are working because they perform approximate integration. Andrew Gordon Wilson has written a lot about this, and I tentatively agree with much of it.
• Your PhD is not your career. As Mark points out, a PhD is just the first step. You'll learn how to do research. You really won't start getting that good at it until a few years in, by which point you'll write up the thesis and start working on something different. You're not even supposed to just keep doing your thesis as you continue your research. The main thing is to have a great research role model, and I think Phillip is quite good (by reputation, I don't know him personally).
• BDL teaches valuable skills. Honestly, I just think statistics is super important for understanding modern deep learning, and it gives you a valuable lens to reason about why things are working. There are other specialisms that can develop valuable skills. But I'd be nervous about trading the opportunity to develop deep familiarity with the stats for practical experience on current SoTA systems (because stats will stay true and important, but SoTA won't stay SoTA). (People I respect disagree with that, and I am uncertain.)

Big picture, I think intellectual diversity among AGI safety researchers is good, Bayesian inference is important and fundamental, and lots of people glom on to whatever the latest hot thing is (currently LLMs), leading to rapid saturation.

So what is interesting to work on? I'm currently thinking about two main things:

• I don't think that exact alignment is possible, in ways that are similar to how exact Bayesian inference is generally possible. So I'm working on trying to learn from the ways in which approximate inference is well/poorly defined to get insights for how alignment can be well/poorly defined and approximated. (Here I agree 100% with Mark that most of what is hard in AGI safety remains framing the problem correctly.)
• I think a huge problem for AGI-esque systems is about to be hunting for dangerous failures. There's a lot of BDL work on 'actively' finding informative data, but mostly for small-data in low-dimensions. I'm much more interested in huge data, high-dimensions, which creates whole new problems (e.g., you can't just compute a score function for each possible datapoint). (Note that this is almost exactly the opposite to Mark's point below! But I don't exactly disagree with him, it's just that lots of things are worth trying.)

There are other things that are important, and I agree that OOD detection is also important (and I'm working on a conceptual paper on this, rather than a detection method specifically). If you'd like to speak about any of this stuff I'm happy to talk. You can reach me at [email protected]

I totally see where you're coming from. I would tend to agree that Bayesian inference doesn't seem to be that useful right now. Visible and exciting leaps are being made by large neural networks, like you say. Bayesian inference on the other hand just works OKish, and faces stiff competition in benchmarks from non-Bayesian methods.

In my opinion, the biggest reasons why Bayesian inference isn't making much of an impact at the moment are:

• Bayes is being applied in simple settings where deep learning works well, and uncertainty is not really needed: large-data, batch training, prediction only tasks. Bayesian inference becomes more important when you have small data, need to update your belief (from heterogeneous sources), and if you need to make decisions on how to act.
• Since we have large datasets, the biggest bottleneck in AI/ML is setting up the problem correctly (i.e. asking the right question). OpenAI models like CLIP and GPT-3, and research directions like multitask/metalearning illustrate this nicely. By setting the problem up in a different way, they can leverage huge amounts of new data. Once you introduce a new problem, you reach for the easiest tool to start to solve it. Bayesian inference is not the easiest tool, so it doesn't contribute much here.
• Current approximate Bayesian training tools don't work very well. (Some disagree with me on this, but I do believe it.)

I also think that in AI alignment, the biggest problem is figuring out the right question to ask. Currently, observable failure cases in the current simple test settings are rare*, which makes it hard to do concrete technical work regardless of whether you choose the to use the Bayesian paradigm.

The thought experiments that motivate AI/ML safety are often longer term, and embedded in larger systems (like society). One place where I do think people have a concrete idea of problems that need solving in AI/ML is social science! I have seen interesting points being made e.g. in FAcct [1] about how you should set up your AI/ML system if you want to avoid certain consequences when deploying in society.

This is where I would focus, if I were to want to work on good outcomes of AI/ML that have impact right now.

Now, I still work on Bayesian ML (although not directly on AI safety). Why do I do this when I agree that Bayesian inference doesn't have much to offer right now? Well, because I think there are reasons to believe that Bayesian inference will have new abilities to offer deep learning (not just uncertainty) in the long term. We just need to work on them! I may turn out to be wrong, but I do still believe that my research direction should be explored in case it turns out to be right. Big companies seem to have large model design under control anyway.

What should you work on? This is difficult. Large model research is difficult to do outside of a few organisations, and requires a lot of engineering effort. You could try to join such an organisation, you could try to investigate properties of trained models, you could try and make these models more widely available, or try to find smaller scale test set-ups where you can probe properties (this is difficult if scale is intrinsically necessary).

I do believe that a great way to get impact right now is to work on questions about deploying ML, and where we want to apply it in society.

You could also remember that your career is longer than a PhD, and work on technical problems now, even if they're not related to AI safety. If you want to work on Bayesian ML, or just ML that quantifies uncertainty in a way that matters (right now), I would work on problems with a decision making element in them. Places where the data-gathering<->decision loop is closed.

Self-driving cars are a cool example, but again difficult to work on outside specific organisations. Bayesian Optimisation or experimental design in industrial settings is a cool small-scale example of where uncertainty modelling really matters. I think smaller scale and lower dimensional applied ML problems are undervalued in the ML community. They're hard to find (it may require talking to people in industry), but I think they are numerous. And what's interesting, is that research can make the difference between a poor solution and a great solution.

I'm curious to hear other people's thoughts on this as well.

[1] https://facctconference.org/

* Correct me if I'm wrong on this. If the failure cases are not rare, then my advice would be to pick one, and try to solve it in a tool agnostic way.

Quantified uncertainty might be fairly important for alignment, since there is a class of approaches that rely on confidence thresholds to avoid catastrophic errors (1, 2, 3). What might also be important is the ability to explicitly control your prior in order to encode assumptions such as those needed for value learning (but maybe there are ways to do it with other methods).

A similar idea is by the way discussed in a post by Jaime Sevilla on the limits of causal discovery: https://towardsdatascience.com/the-limits-of-graphical-causal-discovery-92d92aed54d6

Related to your causality comment above, two days ago I submitted a research proposal on Causal Representation Learning for AI Safety. You may want to see it here: https://www.lesswrong.com/posts/5BkEoJFEqQEWy9GcL/an-open-philanthropy-grant-proposal-causal-representation

Yeah, difficult question... some random thoughts, not sure how helpful:

• for paths where a computational understanding of human cognition is especially useful, expertise with Bayesian modeling of cognition might be promising (and seems already popular)
• relatedly, I have the impression that human cognition does things that would require current approaches to become more Bayesian... e.g. like the things that are worked on under active learning in ML, and models of perception related to active inference in neuroscience
  • this might imply, that we'll (want to) build AI systems that go more into this direction
  • this might also imply, that there are approximations of Bayesian cognition that scale? 
• something something developing AI systems that make explicit predictions about explicit future scenarios might be one of the key applications of highly advanced AI systems and the more "formal" this is, the easier to align?
• you likely talked to him already, but I understood that David Lindner at ETH Zürich does something related to value learning using approaches analogous to Bayesian optimization(?)
• I also wondered how dependend the final design of TAI will be on the current architectures we have today. Even assuming we'll see TAI in less than 20 years, maybe what ends up happening is that TAI will already be largely designed by AI systems? And then maybe more Bayesian architectures are totally on the table if we think it would be helpful?
  • maybe a somewhat useful question to ask: In case AI Safety researchers had a headstart that amounts to multiple years, would making the AI more Bayesian be among the most promising ways to use the headstart?