A library for doing research on developmental interpretability
Project description
DevInterp
A Python Library for Developmental Interpretability Research
DevInterp is a python library for conducting research on developmental interpretability, a novel AI safety research agenda rooted in Singular Learning Theory (SLT). DevInterp proposes tools for detecting, locating, and ultimately controlling the development of structure over training.
Read more about developmental interpretability.
:warning: This library is still in early development. Don't expect things to work on a first attempt. We are actively working on improving the library and adding new features.
Installation
To install devinterp
, simply run pip install devinterp
.
Minimal Example
from devinterp.slt import sample, LLCEstimator
from devinterp.optim import SGLD
from devinterp.utils import optimal_temperature
# Assuming you have a PyTorch Module and DataLoader
llc_estimator = LLCEstimator(..., temperature=optimal_temperature(trainloader))
sample(model, trainloader, ..., callbacks = [llc_estimator])
llc_mean = llc_estimator.sample()["llc/mean"]
Advanced Usage
To see DevInterp in action, check out our example notebooks:
For more advanced usage, see the Diagnostics notebook and for a quick guide on picking hyperparameters, see the Calibration notebook. . Documentation can be found here.
For papers that either inspired or used the DevInterp package, click here.
Known Issues
- The current implementation does not work with transformers out-of-the-box. This can be fixed by adding a wrapper to your model, for example passing Unpack(model) to sample() where unpack is defined by:
class Unpack(nn.Module):
def __init__(model: nn.Module):
self.model = model
def forward(data: Tuple[torch.Tensor, torch.Tensor]):
return self.model(*data)
- LLC Estimation is currently more of an art than a science. It will take some time and pain to get it work reliably.
If you run into issues not mentioned here, please first check the github issues, then ask in the DevInterp Discord, and only then make a new github issue.
Contributing
See CONTRIBUTING.md for guidelines on how to contribute.
Credits & Citations
This package was created by Timaeus. The main contributors to this package are Jesse Hoogland, Stan van Wingerden, and George Wang. Zach Furman, Matthew Farrugia-Roberts and Edmund Lau also made valuable contributions or provided useful advice.
If this package was useful in your work, please cite it as:
@misc{devinterp2024,
title = {DevInterp},
author = {Jesse Hoogland, Stan van Wingerden, and George Wang},
year = {2024},
howpublished = {\url{https://github.com/timaeus-research/devinterp}},
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for devinterp-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e49018c6015bb367494b9a49a4fb52987ceae1b1ed5d6816d5e58670724f0792 |
|
MD5 | 1007f78c206e021aaa5bd93f17f92e02 |
|
BLAKE2b-256 | 69e3df56af6a5f67964aba9ed9ac1bd6e452d52a9d14ba658a2b1bdf8ea5253a |