Alignment research exercises

It’s currently hard to know where to start when trying to get better at thinking about alignment. So below I’ve listed a few dozen exercises which I expect to be helpful. They assume a level of background alignment knowledge roughly equivalent to what’s covered in the technical alignment track of the AGI safety fundamentals course. They vary greatly in difficulty—some are standard knowledge in ML, some are open research questions. I’ve given the exercises star ratings from * to *** for difficulty (note: not for length of time to complete—many require reading papers before engaging with them). However, I haven’t tried to solve them all myself, so the star ratings may be significantly off.

I’ve erred on the side of including exercises which seem somewhat interesting and alignment-related even when I’m uncertain about their value; when working through them, you should keep the question “is this actually useful? Why or why not?” in mind as a meta-exercise. This post will likely be updated over time to remove less useful exercises and add new ones.

I’d appreciate any contributions of:

  1. Comments about which exercises seem most or least useful.

  2. Answers to the exercises.

  3. More exercises! The ideal exercises are nerdsnipe-style problems which can be stated clearly, and seem well-defined, but lead into interesting depths when explored.

Reward learning

  1. * Look at the examples of human feedback mechanisms discussed in the reward-rational implicit choice paper. Think of another type of human feedback. What is the choice set? What is the grounding function?

    1. * This paper by Anthropic introduces a technique called context distillation. Describe this in terms of the reward-rational implicit choice framework.

    2. * Estimate the bandwidth of information conveyed by different types of human feedback. Describe a rough model for how this might change as training progresses. By contrast, how much information is conveyed by the choice of a programmatic reward function? (Consider both the case where the agent is given the exact reward function, and where it learns from reward observations.)

  2. * Look at the examples of biases discussed in learning the preferences of ignorant agents. Identify another bias which similarly influences human decision-making. Describe an example situation where a human with that bias might make the wrong decision. Formulate an algorithm that infers that human’s true preferences.

    1. Some answers here.

  3. ** Given that humans can be assigned any values, why does reward learning ever work in practice?

  4. ** Explain why cooperative inverse reinforcement learning doesn’t solve the alignment problem.

    1. Answer here.

Agency

  1. ** In this paper, researchers devised a test for whether a recurrent network is doing planning: by seeing whether its performance improves when given more time to “think” before it can act. In the AlphaGo paper, researchers compared the performance of their MCTS+neural network algorithm against the network alone. Think of some other test that we could run that would give us evidence about the extent to which some neural network is internally doing planning.

  2. * Consider HCH, an attempted formalisation of “a human’s enlightened judgment”. Why might an implementation of HCH not be aligned? What assumptions would be needed to prevent that?

    1. *** In a later post, Paul defines a stronger version of HCH which “increases the complexity-theoretic expressiveness of HCH. The old version could be computed in EXPTIME, while the new version can compute any decidable function.” Try to rederive a new version of HCH with these properties.

    2. Answer here.

  3. * Ask the OpenAI API about what steps it would take to perform some long-term plan. Work in groups: think of a task that you expect it will be difficult to generate a good plan for, and then see who can design a prompt that will produce the best plan from the API.

    1. * Some steps of a plan generated by the API can also be performed by the API—e.g. a step which requires writing a poem about a given topic. What’s the hardest task you can find for which the API can not only generate a plan, but also perform each of the steps in that plan?

  4. ** Pearl argues that neural networks trained on supervised or self-supervised data can’t learn to reason about interventions and counterfactuals (see this post for an explanation of the distinction). What’s the strongest counterargument against his position?

Reinforcement learning

  1. ** How is supervised learning on reward-maximising trajectories related (mathematically) to policy gradient with sparse, binary rewards?

  2. ** What decision theories are implemented by different RL algorithms?

    1. Some answers here.

  3. ** What might lead an RL agent to learn a policy which sacrifices reward in its current episode to get higher reward in a later episode?

    1. Some answers in section 7 here.

  4. * Self-play in zero-sum two-player games converges to an optimal strategy (given sufficient assumptions about the model class). In other games, this isn’t the case—why not?

  5. ** Evaluate this paper (Reward is Enough). Does their argument hold up?

    1. ** After doing that: consider a bird practicing singing, which listens to its own song, and does RL using the rule: the better the song sounds, the higher the reward. But the bird is also deciding how much time to spend practicing singing versus foraging, etc. And the worse it sings, the more important it is to practice! So you really want the rule: the worse the song sounds, the more rewarding it is to practice singing. How could you resolve this conflict?

    2. Some answers here.

  6. * Why can a behaviourally cloned policy perform well when run for a small set of timesteps, but poorly when run over a longer series of timesteps? How can this be fixed?

    1. Some answers here.

  7. ** If a deep q-learning agent is trained in an environment where some actions lead to large negative rewards, it will never stop trying these actions (the policy will sometimes take these actions even when not randomly exploring due to epsilon exploration). Why does this happen? How could it be prevented?

    1. Some answers here.

  8. ** RL agents have become capable of competent behaviour over longer and longer episodes. What difficulties arise in trying to measure improvements in how long they can act competently for? What metrics are most useful?

    1. The same question, but for sample efficiency rather than episode length.

    2. Some answers here.

Neural networks

  1. * Consider this paper on modularity in neural networks. Evaluate their metric of clustering; what others could we use instead?

  2. ** Consider the following alignment proposal: a neural network has two output heads, one of which chooses actions, the other of which predicts the longer-term consequences of those actions. Suppose that we train the latter head to maximise human-evaluated prediction quality. What differences might we expect from backpropagating that loss all the way through the network, versus only backpropagating through the prediction head? What complications arise if we try to train the prediction head via RL? What advantages might there be of doing so?

  3. ** “Gradient hacking” is a hypothesised phenomenon by which a model decides its actions partly on the basis of its observations of its own parameters, thereby changing the way its parameters are updated. Does the gradient hacking mechanism described in the linked post work? If not, does any variant of it work?

    1. Some answers here.

  4. * Read Jacob Steinhardt’s list of examples of emergent shifts in machine learning. Can you think of any others? What about shifts that you expect in the near future?

  5. ** What might it look like for the circuits hypothesis to be false?

  6. * This paper discusses the metric of “effective data transferred”. What are the limitations of this metric? What are some alternative ways to measure data transfer?

Alignment theory

  1. * Consider extending reinforcement learning to the case where rewards can depend on the parameters of a model. Why do classic convergence proofs no longer work?

    1. *** Are there any limiting assumptions which might lead to interesting theoretical results?

  2. ** One concern with proposals to train using loss functions that depend directly on neural activations is that if we train a network to avoid carrying out any particular piece of cognition, that cognition will instead just be distributed across the network in a way that we can’t detect. Describe a toy example of a cognitive trait that we can currently detect automatically. Design an experiment to determine whether, after training to remove that trait, the network has learned to implement an equivalent trait in a less-easily-detectable way.

  3. *** Rederive some of the proofs in the following papers. For b) and c), explain what assumptions are being made about the optimality of the agents involved, and how they might break down in practice:

    1. Seeking Power is Convergently Instrumental in MDPs

    2. AI safety via debate (and see also proofs about the effects of adding cross-examination)

    3. Alignment proposals and complexity classes (and this follow-up)

    4. Some answers here.

  4. *** Produce a proposal for the ELK prize (note that this requires engaging with the ELK writeup, which is very long).

  5. ** Suppose that we’re training a model via behavioural cloning of a human, but the human starts off with different prior knowledge to the model (either more knowledge, or less knowledge). How might this lead the model to behave in a misaligned way?

    1. Some answers here.

Agent foundations

  1. * Open-source game theory

  2. ** Selection theorems

  3. ** Fixed point exercises

  4. *** The 5 and 10 problem

Evolution and economics

  1. * An old study split insects into several groups which each lived together, and artificially selected in favour of smaller groups, in an attempt to study whether they would evolve to voluntarily restrain their breeding. Predict the outcome of the study.

    1. Some answers here. Did the bias discussed in this post influence your expectations?

  2. ** What might explain why there are so few hermaphroditic animal species, given that every individual being able to bear children could potentially double the number of children in the next generation?

    1. Some answers here.

  3. * Read this post about evolving to extinction. Mathematically demonstrate that segregation-distorters could in fact lead a species to evolve to extinction.

  4. * Evaluate Fletcher and Doebeli’s model of the evolution of altruism.

    1. Use the model to show how the green-beard effect could lead to the evolution of (a certain type of) altruism.

  5. ** Why are roughly equal numbers of males and females born in most species?

    1. Some answers here.

  6. * Comparing GDP across time requires reference to a standard basket of goods and services. What difficulties might this cause in taking GDP comparisons at face value?

    1. Some answers here.

  7. ** Evaluate Roodman’s model of explosive economic growth.

  8. * In cooperative game theory, the “core” is the term for the set of allocations of payoffs to agents where no subset of the agents can form a coalition to improve their payoffs. For example, consider a group of N miners, who have discovered large bars of gold. Assume that two miners can carry one piece of gold, and so the payoff of any coalition S is floor(|S|/​2). If N is even, then the core consists of the single payoff distribution where each miner gets ½. If N is odd, then the core is empty (because the miner who is left out can always make a better offer to some miner who currently has a gold-carrying partner). Identify the core for the following games:

    1. A game with 2001 players: 1000 of them have 1 left shoe, 1001 have 1 right shoe. A left-shoe/​right-shoe pair can be sold for $10.

    2. Mr A and Mr B each have three gloves. Any two gloves make a pair that they can sell for $5.

    3. Answers here.

  9. * How should coalitions decide how to split the payoffs they receive? The concept of Shapley values provides one answer. Convince yourself that Shapley values have the properties of linearity, null player and the stand-alone test described in the linked article.

Some important concepts in ML

These are intended less as exercises and more as pointers to open questions at the cutting edge of deep learning.

  1. Scaling laws

    1. Why do they have the form they do?

    2. Some answers here and here

  2. Neural networks memorisation

  3. Double descent

  4. The lottery ticket hypothesis

  5. Games with spinning-top structure

  6. Gradient noise scale (see also here)

  7. OpenAI Requests for Research

  8. OpenAI Requests for Research 2

Miscellaneous

  1. * Fill in your estimates in Cotra’s timeline model. Does the model broadly make sense to you; are there ways you’d change it?

  2. * Try playing OpenAI’s implementation of the Debate game.

  3. ** Identify an important concept in alignment that isn’t currently very well-explained; write a more accessible explanation.