Memory-augmented QA
Neural QA system focused on compositional questions using attention over key–value facts: attend, retrieve, answer.
This project focuses on developing a memory-augmented question answering (QA) system that leverages external knowledge bases to enhance its ability to answer questions accurately.
The system is designed to retrieve and utilize relevant information from a structured set of facts, improving its performance on various QA tasks. The key components and techniques used in this project include:
- Implemented a Key-Value Memory Network (KVMemNet) to store facts as (key, value) pairs and retrieve relevant values via attention.
- Added batched dot-product attention over keys with value aggregation; supported single-hop reasoning.
- Constructed a tractable dataset through templated questions, vocabulary reduction, and distractor sampling to control difficulty and size.
- Trained the model using CrossEntropy loss, monitored performance metrics, and implemented strategies to mitigate overfitting.
- Improved retrieval by natural-language key formatting, and balanced distractors to encourage fine-grained discrimination to ensure robust performance.