AutoSense Model for Word Sense Induction
This code was used in the experiments of the research paper
Reinald Kim Amplayo, Seung-won Hwang, and Min Song. AutoSense Model for Word Sense Induction. AAAI, 2019.
The src/models folder contains one Java file containing the GAS class. The GAS (Granularity-Agnostic Sense Model) refers to the AutoSense model. To use the model, create an object of GAS using the following line:
GAS gas = new GAS(data, target, numSenses, numTopics, alpha, beta, gamma);
where
data: is a list of data instancestarget: is the target wordnumSenses: is the number of senses hyperparameternumTopics: is the number of topics hyperparameteralpha: is the Dirichlet prior of the topic distribution (set to 0.1 in the paper)beta: is the Dirichlet prior of the sense distribution (set to 0.01 in the paper)gamma: is the Dirichlet prior of the switch distribution (set to 0.3 in the paper)
Then, you would need to run the Gibbs sampler using the following lines of code:
gas.initialize();
gas.estimate(numIters);
where numIters is the number of iterations (set to 2000 in the paper).
To print the results, use the line:
gas.printSemEval(filename, target);
To cite the paper/code, please use this BibTex:
@inproceedings{amplayo2019granularity,
Author = {Reinald Kim Amplayo and Seung-won Hwang and Min Song},
Booktitle = {AAAI},
Location = {Honolulu, HI},
Year = {2019},
Title = {AutoSense Model for Word Sense Induction},
}
If you have questions, send me an email: reinald.kim at ed dot ac dot uk