Original:  The internal representations also perform better on word-in-context (WiC) task.
Revised:  The internal representations have a better ability to distinguish a word's different senses with the word-in-context (WiC) task.
Edit intent: clarity