Traditional KV-MemNNs first preselect a list of candidate KB facts to compose the key-value memory.
one can first detect entity mentions in the question, and include all KB facts that contains with one of those entities as subject into the memory.