Study how a memory augmented AI agent decides which past conversation to keep or drop.
Benchmark a memory selection policy against LoCoMo or LongMemEval style datasets.
Build a transparent memory pipeline that cites its evidence chain before answering.
| mayiwen0212/memchain | astraltenprosper/uncensored-ai-image-video-generator | yespsam/a-share-us-catalyst | |
|---|---|---|---|
| Stars | 96 | 96 | 96 |
| Language | Python | Python | Python |
| Setup difficulty | moderate | easy | moderate |
| Complexity | 4/5 | 2/5 | 3/5 |
| Audience | researcher | general | developer |
Figures from each repo's GitHub metadata at analysis time.
The included example runs on CPU, but the full training setup described in the paper needs eight high end GPUs.
MemChain is a research project that gives AI chat systems a more transparent way of deciding which pieces of conversation history to remember and use when answering a question. Instead of quietly stuffing whatever text a search step retrieves into the AI's context, MemChain produces an explicit, readable trace of its memory decisions before generating an answer. Given a new question and a pool of candidate memories drawn from earlier conversation, the system first figures out what kind of evidence the question actually needs. It then decides, memory by memory, whether to keep it, drop it, refine it, merge it with another, or stop looking further, and it records this as an ordered chain of steps that cite which memories were used at each point. Only a final, compact set of these approved memories is passed along to a separate model that generates the actual answer, keeping the memory selection process separate from and easier to check than the answer itself. The candidate memory pool itself is built only from the question and the conversation's own history, never from the correct answer, which keeps the process honest and testable rather than quietly shaped by hindsight. The repository includes a small runnable example with sample dialogue sessions and questions so you can see the full output format, including the intent analysis, the memory actions taken, the chain of cited steps, and the final memories passed to the answer model. The full training setup described in the paper uses eight high end GPUs and an approach that first fine tunes the system on examples, then further trains it with reinforcement learning, though the private training data, model checkpoints, API keys, and machine specific scripts used for that training are not included here. What is included is the core method code: the memory pool construction logic, the schema definitions, prompt templates, reward calculations, and a small test suite, along with optional support for open source embedding models if you want dense semantic search over memories rather than plain keyword matching. This is an academic code release intended for inspection and extension by other researchers, and it is licensed under the MIT license.
A research implementation that gives AI chat agents an explicit, readable trace of which memories it used before answering.
Mainly Python. The stack also includes Python, PyTorch, Sentence Transformers.
MIT: free to use, modify, and share, including commercially, as long as you keep the copyright notice.
Setup difficulty is rated moderate, with roughly 30min to a first successful run.
Mainly researcher.
This repo across BitVibe Labs
Don't trust strangers blindly. Verify against the repo.