diff --git a/README.md b/README.md index 65fa98473c988ec9c77bb02470b6f32914493fa4..b6c31114b2e86e8309dedb96746c86b01430a261 100644 --- a/README.md +++ b/README.md @@ -1,21 +1 @@ -# Neural Belief Tracking - -Implementation of different uncertainty in Deep Learning methods for improved Dialogue ***Belief*** Tracking. - -## NBT Model - -Slot Utterance Matching Belief Tracker transformer. Using BERT language model fine tuning, a slot utterance matching Attention mechanism and a RNN latent state tracker to generate a latent belief state. Further embedding distance comparison is used for generating model logits. - -## Uncertainty - -### Loss - -* Standard Cross Entropy Loss -* Label Smoothing KL divergence Loss (improved uncertainty) -* Belief Matching (Dirichlet activation function with ELBO Loss) - -### Strategies - -* Temperature Scaling -* Dropout Ensembling -* Model Ensembling +# Code for Paper Knowing What You Know: Calibrating Dialogue Belief StateDistributions via Ensembles \ No newline at end of file