Interesting project! I think that this mechanistic interpretation work does in similar direction: github.com/callummcdoug...