NIWF is defined in Section 3.3. we learn embeddings of various specificity levels for each metric jointly with the model.
for each metric m, we rank the responses in the training data according to that metric and divide it into K = 5 levels of equal size.