Add pipeline tag, library name, and paper/code links
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,16 +1,20 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
| 3 |
language:
|
| 4 |
- en
|
| 5 |
- zh
|
| 6 |
-
|
| 7 |
-
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# MemSifter-4B-Thinking
|
| 11 |
|
| 12 |
**MemSifter** is a lightweight generative session ranker trained with DAPO reinforcement learning. It serves as the core retrieval component of the [MemSifter](https://github.com/plageon/MemSifter) system—an LLM memory retrieval offloading framework based on outcome-driven proxy reasoning.
|
| 13 |
|
|
|
|
|
|
|
| 14 |
Given a user query and a set of candidate conversation sessions (pre-filtered by a dense embedding model), MemSifter-4B-Thinking performs fine-grained reranking to surface the sessions most relevant to the query, which are then passed as context to a downstream chat LLM.
|
| 15 |
|
| 16 |
## System Pipeline
|
|
@@ -24,6 +28,11 @@ Session Embedding → Session Ranking (MemSifter) → Chat LLM
|
|
| 24 |
2. **Session Ranking** — MemSifter (this model) performs fine-grained reranking of the pre-filtered candidates.
|
| 25 |
3. **Chat LLM** — the top-ranked sessions are assembled into a context window and passed to any OpenAI-compatible chat model.
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
## How to Use
|
| 28 |
|
| 29 |
Install the required packages:
|
|
|
|
| 1 |
---
|
| 2 |
+
base_model:
|
| 3 |
+
- Qwen/Qwen3-4B-Thinking-2507
|
| 4 |
language:
|
| 5 |
- en
|
| 6 |
- zh
|
| 7 |
+
license: mit
|
| 8 |
+
pipeline_tag: text-generation
|
| 9 |
+
library_name: transformers
|
| 10 |
---
|
| 11 |
|
| 12 |
# MemSifter-4B-Thinking
|
| 13 |
|
| 14 |
**MemSifter** is a lightweight generative session ranker trained with DAPO reinforcement learning. It serves as the core retrieval component of the [MemSifter](https://github.com/plageon/MemSifter) system—an LLM memory retrieval offloading framework based on outcome-driven proxy reasoning.
|
| 15 |
|
| 16 |
+
This model is presented in the paper [MemSifter: Offloading LLM Memory Retrieval via Outcome-Driven Proxy Reasoning](https://huggingface.co/papers/2603.03379).
|
| 17 |
+
|
| 18 |
Given a user query and a set of candidate conversation sessions (pre-filtered by a dense embedding model), MemSifter-4B-Thinking performs fine-grained reranking to surface the sessions most relevant to the query, which are then passed as context to a downstream chat LLM.
|
| 19 |
|
| 20 |
## System Pipeline
|
|
|
|
| 28 |
2. **Session Ranking** — MemSifter (this model) performs fine-grained reranking of the pre-filtered candidates.
|
| 29 |
3. **Chat LLM** — the top-ranked sessions are assembled into a context window and passed to any OpenAI-compatible chat model.
|
| 30 |
|
| 31 |
+
## Resources
|
| 32 |
+
|
| 33 |
+
- **GitHub:** [https://github.com/plageon/MemSifter](https://github.com/plageon/MemSifter)
|
| 34 |
+
- **Paper:** [https://arxiv.org/abs/2603.03379](https://arxiv.org/abs/2603.03379)
|
| 35 |
+
|
| 36 |
## How to Use
|
| 37 |
|
| 38 |
Install the required packages:
|