David Bau

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/

Locating and editing factual associations in GPT (NeurIPS 2022)

Mass-editing thousands of facts into a transformer memory (ICLR 2023)

Rewriting a Deep Generative Model, ECCV 2020 (oral). Interactive tool to directly edit the rules of a GAN to synthesize scenes with objects added, removed, or altered. Change StyleGANv2 to make extravagant eyebrows, or horses wearing hats.