This repository has the code for the paper 'Scaling up Discovery of Latent Concepts in Deep NLP Models', Majd Hawasly, Fahim Dalvi, Nadir Durrani, EACL 2024
comparing clustering algorithms (namely, agglomerative hierarchical clustering, leaders algorithm and K-Means) for the purpose of supporting large-scale latent concept discovery in NLP models.
The ConceptX library was used to generate activations per layer for BERT models. After installing the required dependencies of ConceptX, the following command can be used to extract acivations of a specifc layer:
/bin/bash get_activations.sh <ConceptX_SCRIPT_DIR> <PATH_TO_SENTENCE_FILE> <NAME_SENTENCE_FILE> <BERT_MODEL> <MAX_SENTENCE_LENGTH> <MIN_WORD_FREQ> <MAX_WORD_FREQ> <DELETE_FREQ> <TARGET_LAYER>
This script creates a directory for the target layer and a number of files. The required point and vocab numpy files needed for clustering will be created in that folder.
Scripts create_{agglomerative,kmeans,leaders}.py can be used to create the clustering text files.
The scripts require points.npy and vocab.npy of the activations, in addition to an output path and a number of desired clusters.
Refer to the individual scripts for further details on usage.
After the clustering text files are produced, alignment and coverage with regard to human-defined concepts can be computed using the script alignment.py.
Example usage:
python alignment.py --sentence-file <SENTENCES> --label-file <LABELS> --cluster-file <CLUSTERING_RESULT> --threshold <THRESHOLD> --method M2
The script requires a sentence file and and its annotation label file that represent the human concept (e.g. Part of Speech tags).
Threshold is the percentage \theta at which an encoded cluster and a human-defined concept are assumed aligned. In our experiments we used \theta = 0.95.
Please cite this paper if you use the software
@article{hawasly2024scaling,
title={Scaling up Discovery of Latent Concepts in Deep NLP Models,
author={Hawasly, Majd
and Dalvi, Fahim
and Durrani, Nadir},
journal={Proceedings of the The 18th Conference of the European Chapter of the Association for Computational Linguistics (EACL)},
year={2024}
}