Skip to content

TheanLim/fast-influence-functions

Repository files navigation

FastIF with improved search heuristic

This is a work in progress to replicate and improve FastIF search heuristic.
Codes are heavily annotated to facilitate understandings. Link to the Original Paper

Recall of kNN

Replication of the original method: recall_feat_L2
Improved Heuristic: recall_predFeat_IP

Explainability of Influential Examples

Replication of the original method: imitator_feat_L2
Improved Heuristic: imitator_predFeat_IP

Error Correction

MultiNLI→ HANS→ HANS

Replication of the original method: error_correction_hans_feat_L2
Improved Heuristic: error_correction_hans_predFeat_IP

MultiNLI→ MultiNLI→ HANS

Replication of the original method: error_correction_mnli_feat_L2
Improved Heuristic: error_correction_mnli_predFeat_IP

Requirements

Please see requirements.txt for detailed dependencies. The major ones include

  • python 3.6 or later (for type annotations and f-string)
  • pytorch==1.5.1
  • transformers==3.0.2

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published