MOSAIC Code

HKUST KnowComp · long-term conversational memory

Safe memory is not one score.
It fails in four places.

MOSAIC asks whether an assistant can remember what a person said, showed, and conveyed across weeks of conversation. The benchmark holds the question, reader, prompt, and budget fixed, then intervenes on one stage at a time.

923questions
784answerable
139refusals
11–24sessions each

Four stages, one question.

Existing memory benchmarks mostly cover text and images, and they report a single end-to-end number. That number cannot say whether the evidence was lost when it was written down, when it was retrieved, when it was packed into the prompt, or when the model read it.

85.2% of answerable questions need two or more facts. 75.0% need two or more modalities. The media carry the decision: spatial layout, fine visual detail, video order and repetition, or vocal prosody.

  1. 01WriteHow the history is stored: native media, or a text substitute.
  2. 02RetrieveWhether the right evidence is found in the long history.
  3. 03DeliverWhether that evidence actually enters the model request.
  4. 04ReadWhether the reader answers from the evidence it was given.

What is in the release

Point MOSAIC_DATA_ROOT at the downloaded dataset. The evaluation code reads these paths.

Questions

dataset/stage2_human_filtered/

Each item has a question, an evidence chain, and a timestamped session history.

Media

dataset/media/

Images, audio, and video. Stored paths that contain clean_output/media/ are mapped onto this folder.

Text views

textualized_caption/ and textualized_omni/

The same histories as captions, so a text-only system can be compared with native media.

Indexes and records

indexes/ and records/protocol_v1/

RAG, A-MEM, and MemoryOS indexes, plus the paper’s evaluation records.

Evidence roles

Refusal items remove the required evidence. A system should answer when the chain is present and refuse when it is not. Track A holds the reader fixed and is the diagnostic comparison. Track B changes the whole system and is a deployment-style illustration.

What the diagnosis shows

These contrasts keep the question set fixed. Spatial evidence and delivery are the strongest results. Video timing and raw audio versus a transcript are smaller and should not be read as equally settled.

Deliver

Retrieved is not delivered.

On the same top-50 ranking, tightening what enters the request moves the score from 0.708 to 0.670. Complete-chain delivery falls from 0.675 to 0.491. Packing after retrieval is its own failure.

Write · image

Object lists drop the relation.

On the same items, raw images score 32.0 points above OCR. A spatial description scores 15.4 points above a list of objects. Presence of an object is not the layout between objects.

Budget

The tight budget is joint.

The most restrictive setting in the delivery figure changes top-k and the per-modality attachment limit together. The 17.7-point drop belongs to that combined budget, not to the one-media cap alone.

Leaderboard

Paper baselines only. A new row is added to docs/leaderboard.json after the run is checked against the 923-question set and the same judge.

923 questions. Each system uses its own reader and memory stack. Answerable score is the mean judge score on 784 answerable items. Correct refusal is the rate of refusing the 139 items whose evidence was removed. False refusal is the rate of refusing an answerable item. A high correct-refusal rate with a high false-refusal rate is indiscriminate abstention, not a better memory.

#SystemMemoryReader AnswerableCorrect refusal False refusalLatency (s)
1ReActNative multimodalMiMo-V2.50.7320.4100.01873.6
2Naive RAGNative multimodalGemini 3.50.6870.4960.02231.2
3ReActNative multimodalGemini 3.50.6740.4170.03746.4
4A-MEMCaption textDeepSeek-v4-Flash0.6230.7190.173—
5Naive RAGNative multimodalMiMo-V2.50.6140.7340.14329.6
6A-MEMOmni textDeepSeek-v4-Flash0.6110.7630.177—
7ReActOmni textDeepSeek-v4-Flash0.4190.7990.39017.5
8ReActCaption textDeepSeek-v4-Flash0.4150.8200.40418.6
9Naive RAGCaption textDeepSeek-v4-Flash0.3900.8350.4315.8
10Naive RAGOmni textDeepSeek-v4-Flash0.3750.8710.4216.2
11MemoryOSOmni textDeepSeek-v4-Flash0.2550.8920.614228.4
12MemoryOSCaption textDeepSeek-v4-Flash0.2430.8850.649205.4

784 answerable questions. The reader is Gemini 3.5 for every row, so the score isolates the memory backend. Chain coverage is the fraction of items whose full required evidence chain is delivered. MemoryOS coverage is not observable in this run.

#BackendMemory Chain coverageScore
1A-MEMCaption text0.7870.684
2A-MEMOmni text0.7020.664
3Flat RAGNative multimodal0.4660.641
4Flat RAGCaption text0.6980.522
5Flat RAGOmni text0.7050.500
6MemoryOSOmni text—0.279
7MemoryOSCaption text—0.268

Submit a result

Use the released 923 questions and the paper’s judge. Report the answerable score, and for an end-to-end run also the correct-refusal and false-refusal rates. Accepted rows are written into the leaderboard file.

The issue is filed on HKUST-KnowComp/MOSAIC. Include enough detail for someone else to rerun the number.