Merge feat/temporal-reranking: re-ranking keeps the retrieval score; recency metric
CI / test (push) Failing after 2s
CI / test (push) Failing after 2s
Co-Authored-By: Claude Opus 5 (1M context) <[email protected]>
This commit is contained in:
@@ -653,6 +653,46 @@ add. There is no case here for changing the default; `TokenFilter::Stemmed`
|
||||
is available via `HDF5Memory::set_token_filter` for callers who want Hit@5/@10
|
||||
over rank-1 precision.
|
||||
|
||||
### Re-ranking and recency — full haystack, n=500
|
||||
|
||||
`reranker::rerank` combines temporal decay, source authority and Hebbian
|
||||
activation. Until now its combined score contained **no relevance term at
|
||||
all** — `RerankInput` did not carry the retrieval score — so a caller that
|
||||
re-ranked its candidates threw the retriever's ordering away and returned them
|
||||
ordered by age. The OpenClaw backend did exactly that on every search.
|
||||
|
||||
Measuring that is unambiguous. "Recency" below is the share of
|
||||
`knowledge-update` questions where the newest gold session outranked the stale
|
||||
one (see `newest_gold_first`); ~45% is chance.
|
||||
|
||||
| Mode | Hit@1 | Hit@5 | Hit@10 | MRR | recency |
|
||||
|---|---|---|---|---|---|
|
||||
| Hybrid 0.4/0.6, no re-rank | 51.6% | **81.4%** | 87.8% | 0.6430 | 45.0% |
|
||||
| + re-rank, **metadata only** (pre-fix) | 11.0% | 24.8% | 43.8% | 0.1829 | **87.5%** |
|
||||
| + re-rank, relevance-led, half-life 1 day | **52.0%** | 79.8% | 87.8% | 0.6403 | 51.7% |
|
||||
| + re-rank, relevance-led, half-life 7 days | 51.8% | 80.8% | 87.6% | **0.6437** | **52.2%** |
|
||||
| + re-rank, relevance-led, half-life 30 days | 51.8% | 81.0% | 87.8% | 0.6427 | 51.4% |
|
||||
| + re-rank, relevance-led, half-life 90 days | **52.0%** | 80.4% | 87.8% | 0.6425 | 50.8% |
|
||||
|
||||
**The pre-fix row is the finding.** Ordering candidates by recency alone costs
|
||||
40.6pp of Hit@1 and two thirds of MRR: the results are the newest memories in
|
||||
the pool rather than the ones that answer the question. It does ace the recency
|
||||
metric, which is exactly what makes that metric worth having — a number that
|
||||
only goes up when a change is good would not have caught this.
|
||||
|
||||
With relevance leading, retrieval is preserved (Hit@1 +0.4pp, MRR −0.003
|
||||
against no re-ranking) and recency discrimination gains 6–7pp. That is a real
|
||||
improvement but not a solved problem: recency only breaks near-ties, so it
|
||||
cannot reach the 87.5% the degenerate ordering gets. Those two rows are the
|
||||
ends of a trade-off, and the default sits deliberately near the relevance end.
|
||||
|
||||
**Half-life is not a sensitive knob.** Across 1, 7, 30 and 90 days recency
|
||||
moves 1.4pp and MRR 0.003 — inside the noise of a 500-question run — because
|
||||
the temporal term is capped by its weight (0.3) while relevance differences
|
||||
between candidates are larger. The 24-hour default is kept; there is no
|
||||
measured reason to change it, and a corpus-matched value is not the lever it
|
||||
looks like.
|
||||
|
||||
### Weight sweep — full haystack, n=500
|
||||
|
||||
`0.7/0.3` was a documented default, never a searched one. Sweeping
|
||||
|
||||
@@ -1,5 +1,28 @@
|
||||
# Changelog
|
||||
|
||||
## Unreleased
|
||||
|
||||
### Retrieval quality
|
||||
- `clawhdf5-agent`: **re-ranking discarded the retrieval score.**
|
||||
`reranker::rerank` built its combined score from temporal decay, source
|
||||
authority and Hebbian activation only — `RerankInput` had no relevance field
|
||||
— so re-ranking a candidate pool reordered it by age and threw the
|
||||
retriever's ordering away. The OpenClaw backend re-ranked every search, so
|
||||
this was its shipping behaviour: measured over the full LongMemEval haystack
|
||||
it cost **40.6pp of Hit@1** (11.0% vs 51.6%) and two thirds of MRR (0.183 vs
|
||||
0.643). `RerankInput::relevance` and `ReRankConfig::relevance_weight` (1.0 by
|
||||
default) fix it: relevance leads and the metadata signals break near-ties,
|
||||
which restores retrieval (Hit@1 +0.4pp vs no re-ranking) and improves
|
||||
recency discrimination by 6–7pp. **Breaking:** `RerankInput` and
|
||||
`ReRankConfig` gained fields, so literal constructions need updating;
|
||||
`..Default::default()` does not.
|
||||
- `clawhdf5-bench`: the LongMemEval harness feeds the dataset's real session
|
||||
dates to the store instead of a synthetic counter (decay needs true
|
||||
intervals, not just the right order), and reports `newest_gold_first` — on a
|
||||
`knowledge-update` question, did the newest gold session outrank the stale
|
||||
one it supersedes? Plain recall cannot see this, because both are labelled
|
||||
gold. New `--rerank-sweep`.
|
||||
|
||||
## v2.5.0 (2026-09-19)
|
||||
|
||||
### Upgrade Notes
|
||||
|
||||
@@ -554,6 +554,7 @@ impl MemoryBackend for ClawhdfBackend {
|
||||
timestamp: r.timestamp,
|
||||
source_channel: r.source_channel.clone(),
|
||||
raw_activation: r.activation,
|
||||
relevance: r.score,
|
||||
})
|
||||
.collect();
|
||||
|
||||
|
||||
@@ -4,8 +4,10 @@
|
||||
//! into a single composite score for each retrieved result.
|
||||
|
||||
/// Configuration for the multi-factor re-ranker.
|
||||
#[derive(Debug, Clone)]
|
||||
#[derive(Debug, Clone, Copy)]
|
||||
pub struct ReRankConfig {
|
||||
/// Weight applied to the retrieval score the candidate arrived with.
|
||||
pub relevance_weight: f32,
|
||||
/// Weight applied to the temporal decay score (0.0–1.0).
|
||||
pub temporal_weight: f32,
|
||||
/// Weight applied to the source authority score (0.0–1.0).
|
||||
@@ -20,6 +22,9 @@ pub struct ReRankConfig {
|
||||
impl Default for ReRankConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
// Relevance leads: the metadata signals break ties and nudge, they
|
||||
// do not decide. See `BENCHMARKS.md`, "Recency discrimination".
|
||||
relevance_weight: 1.0,
|
||||
temporal_weight: 0.3,
|
||||
authority_weight: 0.2,
|
||||
activation_weight: 0.5,
|
||||
@@ -41,6 +46,8 @@ pub struct ReRankResult {
|
||||
pub authority_score: f32,
|
||||
/// Normalised Hebbian activation score in [0, 1].
|
||||
pub activation_score: f32,
|
||||
/// The retrieval score carried through from the input.
|
||||
pub relevance_score: f32,
|
||||
}
|
||||
|
||||
/// Compute an exponential decay temporal score.
|
||||
@@ -105,6 +112,15 @@ pub struct RerankInput {
|
||||
pub source_channel: String,
|
||||
/// Raw Hebbian activation weight for this entry.
|
||||
pub raw_activation: f32,
|
||||
/// The retrieval score that put this entry in the candidate list.
|
||||
///
|
||||
/// Re-ranking is meant to *adjust* the retriever's ordering with signals
|
||||
/// it does not have, not to replace it. Without this the combined score
|
||||
/// was made of recency, authority and activation alone, so a candidate
|
||||
/// pool came back ordered by age with its relevance ordering discarded.
|
||||
/// Callers with no meaningful score can pass the same value for every
|
||||
/// entry, which reduces to the old behaviour.
|
||||
pub relevance: f32,
|
||||
}
|
||||
|
||||
/// Re-rank a list of retrieval results using multi-factor scoring.
|
||||
@@ -138,7 +154,8 @@ pub fn rerank(
|
||||
let auth = source_authority_score(&inp.source_channel);
|
||||
let act = activation_score(inp.raw_activation);
|
||||
|
||||
let combined = config.temporal_weight * ts
|
||||
let combined = config.relevance_weight * inp.relevance
|
||||
+ config.temporal_weight * ts
|
||||
+ config.authority_weight * auth
|
||||
+ config.activation_weight * act;
|
||||
|
||||
@@ -148,6 +165,7 @@ pub fn rerank(
|
||||
temporal_score: ts,
|
||||
authority_score: auth,
|
||||
activation_score: act,
|
||||
relevance_score: inp.relevance,
|
||||
}
|
||||
})
|
||||
.collect();
|
||||
@@ -253,22 +271,51 @@ mod tests {
|
||||
timestamp: 0.0, // very old
|
||||
source_channel: "other".to_string(),
|
||||
raw_activation: 0.1,
|
||||
relevance: 0.0,
|
||||
},
|
||||
RerankInput {
|
||||
index: 1,
|
||||
timestamp: 86_400.0, // one day ago
|
||||
source_channel: "conversation".to_string(),
|
||||
raw_activation: 0.5,
|
||||
relevance: 0.0,
|
||||
},
|
||||
RerankInput {
|
||||
index: 2,
|
||||
timestamp: 172_800.0, // "now"
|
||||
source_channel: "user_correction".to_string(),
|
||||
raw_activation: 1.0,
|
||||
relevance: 0.0,
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn relevance_leads_but_recency_breaks_near_ties() {
|
||||
let entry = |index, timestamp, relevance| RerankInput {
|
||||
index,
|
||||
timestamp,
|
||||
source_channel: "conversation".to_string(),
|
||||
raw_activation: 1.0,
|
||||
relevance,
|
||||
};
|
||||
let now = 10.0 * 86_400.0;
|
||||
let config = ReRankConfig::default();
|
||||
|
||||
// A clearly better match wins despite being much older. Before
|
||||
// `relevance` existed the combined score ignored it entirely, so this
|
||||
// returned the newer, irrelevant entry.
|
||||
let ranked = rerank(&[entry(0, 0.0, 1.0), entry(1, now, 0.1)], &config, now);
|
||||
assert_eq!(ranked[0].index, 0, "{ranked:?}");
|
||||
|
||||
// Between near-equal matches, the newer one wins.
|
||||
let ranked = rerank(&[entry(0, 0.0, 0.80), entry(1, now, 0.79)], &config, now);
|
||||
assert_eq!(ranked[0].index, 1, "{ranked:?}");
|
||||
|
||||
// The breakdown carries the relevance through.
|
||||
assert_eq!(ranked[0].relevance_score, 0.79);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn rerank_returns_all_entries() {
|
||||
let inputs = make_inputs();
|
||||
@@ -302,6 +349,7 @@ mod tests {
|
||||
#[test]
|
||||
fn rerank_score_breakdown_matches_manual_calculation() {
|
||||
let config = ReRankConfig {
|
||||
relevance_weight: 0.0,
|
||||
temporal_weight: 1.0,
|
||||
authority_weight: 0.0,
|
||||
activation_weight: 0.0,
|
||||
@@ -312,6 +360,7 @@ mod tests {
|
||||
timestamp: 0.0,
|
||||
source_channel: "other".to_string(),
|
||||
raw_activation: 0.5,
|
||||
relevance: 0.0,
|
||||
}];
|
||||
let now = 3600.0_f64; // exactly one half-life later
|
||||
let results = rerank(&inputs, &config, now);
|
||||
|
||||
@@ -57,7 +57,8 @@ mod embedder;
|
||||
|
||||
use clawhdf5_agent::bm25::TokenFilter;
|
||||
use clawhdf5_agent::hybrid::Fusion;
|
||||
use clawhdf5_agent::{AgentMemory, HDF5Memory, MemoryConfig, MemoryEntry};
|
||||
use clawhdf5_agent::reranker::{ReRankConfig, RerankInput, rerank};
|
||||
use clawhdf5_agent::{AgentMemory, HDF5Memory, MemoryConfig, MemoryEntry, SearchResult};
|
||||
use serde::Deserialize;
|
||||
use tempfile::TempDir;
|
||||
|
||||
@@ -69,9 +70,17 @@ fn describe(mode: Mode) -> String {
|
||||
Fusion::Weighted { vector, keyword } => format!("vector_{vector:.1}_keyword_{keyword:.1}"),
|
||||
Fusion::Rrf { k } => format!("rrf_k{k:.0}"),
|
||||
};
|
||||
match mode.tokens {
|
||||
let tokens = match mode.tokens {
|
||||
TokenFilter::Plain => fusion,
|
||||
TokenFilter::Stemmed => format!("{fusion}_stemmed"),
|
||||
};
|
||||
match mode.rerank {
|
||||
None => tokens,
|
||||
Some(cfg) if cfg.relevance_weight == 0.0 => format!("{tokens}_rerank_metadata"),
|
||||
Some(cfg) => format!(
|
||||
"{tokens}_rerank_blended_hl{:.0}d",
|
||||
cfg.temporal_half_life_secs / 86_400.0
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -83,6 +92,9 @@ struct Mode {
|
||||
fusion: Fusion,
|
||||
/// How keyword tokens are normalised before indexing and querying.
|
||||
tokens: TokenFilter,
|
||||
/// Re-rank the retrieved candidates with recency and friends, relative to
|
||||
/// the question's own date.
|
||||
rerank: Option<ReRankConfig>,
|
||||
}
|
||||
|
||||
impl Mode {
|
||||
@@ -91,9 +103,17 @@ impl Mode {
|
||||
label,
|
||||
fusion: Fusion::Weighted { vector, keyword },
|
||||
tokens: TokenFilter::Plain,
|
||||
rerank: None,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg_attr(not(feature = "embeddings"), allow(dead_code))]
|
||||
fn reranked(mut self, label: &'static str, rerank: ReRankConfig) -> Self {
|
||||
self.label = label;
|
||||
self.rerank = Some(rerank);
|
||||
self
|
||||
}
|
||||
|
||||
const fn stemmed(mut self, label: &'static str) -> Self {
|
||||
self.label = label;
|
||||
self.tokens = TokenFilter::Stemmed;
|
||||
@@ -121,11 +141,61 @@ const RRF: Mode = Mode {
|
||||
label: "Hybrid (reciprocal rank fusion, k=60)",
|
||||
fusion: Fusion::Rrf { k: 60.0 },
|
||||
tokens: TokenFilter::Plain,
|
||||
rerank: None,
|
||||
};
|
||||
|
||||
/// The same two configurations with stemmed keyword tokens, so the tokenizer's
|
||||
/// effect is isolated from everything else.
|
||||
const BM25_STEMMED: Mode = BM25_ONLY.stemmed("BM25 only, stemmed tokens");
|
||||
|
||||
/// Re-ranking as it behaved before `relevance` was an input: the combined
|
||||
/// score was recency + authority + activation only, so the retriever's own
|
||||
/// ordering was discarded.
|
||||
#[cfg(feature = "embeddings")]
|
||||
fn hybrid_rerank_metadata_only() -> Mode {
|
||||
HYBRID.reranked(
|
||||
"Hybrid + rerank (metadata only, pre-fix)",
|
||||
ReRankConfig {
|
||||
relevance_weight: 0.0,
|
||||
..ReRankConfig::default()
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
/// Re-ranking as it behaves now: relevance leads, recency nudges.
|
||||
#[cfg(feature = "embeddings")]
|
||||
fn hybrid_rerank_blended() -> Mode {
|
||||
HYBRID.reranked(
|
||||
"Hybrid + rerank (relevance + recency)",
|
||||
ReRankConfig::default(),
|
||||
)
|
||||
}
|
||||
|
||||
/// The same blend at several half-lives. Decay is `2^(-age / half_life)`, so a
|
||||
/// half-life far shorter than the gaps between memories sends every score to
|
||||
/// zero and the signal vanishes; far longer and everything scores ~1 and it
|
||||
/// vanishes the other way. The right value tracks how far apart the memories
|
||||
/// actually are.
|
||||
#[cfg(feature = "embeddings")]
|
||||
fn hybrid_rerank_half_lives() -> Vec<Mode> {
|
||||
[
|
||||
("1 day", 86_400.0),
|
||||
("7 days", 7.0 * 86_400.0),
|
||||
("30 days", 30.0 * 86_400.0),
|
||||
("90 days", 90.0 * 86_400.0),
|
||||
]
|
||||
.into_iter()
|
||||
.map(|(label, half_life)| {
|
||||
HYBRID.reranked(
|
||||
Box::leak(format!("Hybrid + rerank, half-life {label}").into_boxed_str()),
|
||||
ReRankConfig {
|
||||
temporal_half_life_secs: half_life,
|
||||
..ReRankConfig::default()
|
||||
},
|
||||
)
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
#[cfg(feature = "embeddings")]
|
||||
const HYBRID_STEMMED: Mode = HYBRID.stemmed("Hybrid 0.4/0.6, stemmed tokens");
|
||||
|
||||
@@ -217,6 +287,37 @@ struct Question {
|
||||
haystack_session_ids: Vec<String>,
|
||||
haystack_sessions: Vec<Vec<Turn>>,
|
||||
answer_session_ids: Vec<String>,
|
||||
/// One timestamp per haystack session, e.g. "2023/05/25 (Thu) 20:21".
|
||||
#[serde(default)]
|
||||
haystack_dates: Vec<String>,
|
||||
}
|
||||
|
||||
/// Seconds since the epoch for a LongMemEval session date, which looks like
|
||||
/// `2023/05/25 (Thu) 20:21`. Sessions are stored in chronological order, so a
|
||||
/// date that cannot be parsed falls back to its position — order is preserved
|
||||
/// even if the interval is not.
|
||||
fn session_time(date: &str, position: usize) -> f64 {
|
||||
let stamp = |y: i64, mo: i64, d: i64, h: i64, mi: i64| -> f64 {
|
||||
// Days since 1970-01-01 via the civil-from-days algorithm.
|
||||
let (y, mo) = if mo <= 2 { (y - 1, mo + 12) } else { (y, mo) };
|
||||
let era = y.div_euclid(400);
|
||||
let yoe = y - era * 400;
|
||||
let doy = (153 * (mo - 3) + 2) / 5 + d - 1;
|
||||
let doe = yoe * 365 + yoe / 4 - yoe / 100 + doy;
|
||||
let days = era * 146_097 + doe - 719_468;
|
||||
(days * 86_400 + h * 3_600 + mi * 60) as f64
|
||||
};
|
||||
let parse = || -> Option<f64> {
|
||||
let (ymd, rest) = date.split_once(' ')?;
|
||||
let mut ymd = ymd.split('/');
|
||||
let y = ymd.next()?.parse().ok()?;
|
||||
let mo = ymd.next()?.parse().ok()?;
|
||||
let d = ymd.next()?.parse().ok()?;
|
||||
let hm = rest.rsplit(' ').next()?;
|
||||
let (h, mi) = hm.split_once(':')?;
|
||||
Some(stamp(y, mo, d, h.parse().ok()?, mi.parse().ok()?))
|
||||
};
|
||||
parse().unwrap_or(1_000_000.0 + position as f64 * 86_400.0)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
@@ -235,11 +336,21 @@ struct Metrics {
|
||||
rr_turn: f64,
|
||||
abstention_correct: u32,
|
||||
abstention_total: u32,
|
||||
/// Questions where the newest gold session outranked the older ones, out
|
||||
/// of those with more than one gold session and at least one retrieved.
|
||||
newest_gold_first: u32,
|
||||
newest_gold_total: u32,
|
||||
latency_ns: Vec<u64>,
|
||||
count: u32,
|
||||
}
|
||||
|
||||
impl Metrics {
|
||||
/// `None` when no question in this bucket had multiple gold sessions.
|
||||
fn newest_gold_first_pct(&self) -> Option<f64> {
|
||||
(self.newest_gold_total > 0)
|
||||
.then(|| self.newest_gold_first as f64 / self.newest_gold_total as f64 * 100.0)
|
||||
}
|
||||
|
||||
fn hit1_session_pct(&self) -> f64 {
|
||||
self.hit1_session as f64 / self.count.max(1) as f64 * 100.0
|
||||
}
|
||||
@@ -297,6 +408,16 @@ struct EvalResult {
|
||||
hit5_turn: bool,
|
||||
hit10_turn: bool,
|
||||
rr_turn: Option<f64>,
|
||||
/// For a question whose evidence spans several dated sessions (a
|
||||
/// `knowledge-update`, where an earlier fact is superseded by a later
|
||||
/// one): did the *newest* gold session outrank every older gold session
|
||||
/// that was returned? `None` when the question has one gold session, or
|
||||
/// when none were retrieved, so there is nothing to discriminate.
|
||||
///
|
||||
/// Plain recall cannot see this. LongMemEval labels *both* the stale and
|
||||
/// the updated session as gold, so returning either counts as a hit — yet
|
||||
/// only one of them answers the question correctly.
|
||||
newest_gold_first: Option<bool>,
|
||||
latency: Duration,
|
||||
}
|
||||
|
||||
@@ -317,15 +438,21 @@ fn evaluate_question(
|
||||
// Build MemoryEntry list from all haystack sessions
|
||||
let mut entries: Vec<MemoryEntry> = Vec::new();
|
||||
let mut turn_has_answer: Vec<bool> = Vec::new();
|
||||
let mut ts = 1_000_000.0f64;
|
||||
|
||||
for (sess_idx, session) in q.haystack_sessions.iter().enumerate() {
|
||||
let sess_id = q
|
||||
.haystack_session_ids
|
||||
.get(sess_idx)
|
||||
.map(String::as_str)
|
||||
.unwrap_or("unknown");
|
||||
for turn in session {
|
||||
// Real session dates, not a synthetic counter: anything that decays
|
||||
// with age needs true intervals, not just the right order.
|
||||
let session_start = q
|
||||
.haystack_dates
|
||||
.get(sess_idx)
|
||||
.map_or(sess_idx as f64 * 86_400.0, |d| session_time(d, sess_idx));
|
||||
for (turn_idx, turn) in session.iter().enumerate() {
|
||||
// Spread a session's turns over the minutes following its start.
|
||||
let ts = session_start + turn_idx as f64 * 60.0;
|
||||
entries.push(MemoryEntry {
|
||||
chunk: turn.content.clone(),
|
||||
embedding: embedding_for(embeddings, &turn.content),
|
||||
@@ -339,7 +466,6 @@ fn evaluate_question(
|
||||
},
|
||||
});
|
||||
turn_has_answer.push(turn.has_answer);
|
||||
ts += 1.0;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -356,11 +482,87 @@ fn evaluate_question(
|
||||
// Set of session IDs that contain the answer
|
||||
let answer_sess_set: HashSet<&str> = q.answer_session_ids.iter().map(String::as_str).collect();
|
||||
|
||||
// When each gold session was recorded, so "newest" is by date rather than
|
||||
// by position (the two agree in this dataset, but the metric should not
|
||||
// depend on that).
|
||||
let gold_times: HashMap<&str, f64> = q
|
||||
.haystack_session_ids
|
||||
.iter()
|
||||
.enumerate()
|
||||
.filter(|(_, sid)| answer_sess_set.contains(sid.as_str()))
|
||||
.map(|(i, sid)| {
|
||||
let t = q
|
||||
.haystack_dates
|
||||
.get(i)
|
||||
.map_or(i as f64 * 86_400.0, |d| session_time(d, i));
|
||||
(sid.as_str(), t)
|
||||
})
|
||||
.collect();
|
||||
|
||||
let query_emb = embedding_for(embeddings, &q.question);
|
||||
let t0 = Instant::now();
|
||||
let results = memory.hybrid_search_with(&query_emb, &q.question, mode.fusion, top_k);
|
||||
// Re-ranking only reorders; it needs a candidate pool larger than `top_k`
|
||||
// to have anything to promote.
|
||||
let pool = if mode.rerank.is_some() {
|
||||
top_k * 4
|
||||
} else {
|
||||
top_k
|
||||
};
|
||||
let mut results = memory.hybrid_search_with(&query_emb, &q.question, mode.fusion, pool);
|
||||
if let Some(config) = mode.rerank {
|
||||
// "Now" is the moment the question was asked, so decay measures how
|
||||
// stale each memory was at that point.
|
||||
let now = session_time(&q.question_date, q.haystack_sessions.len());
|
||||
let inputs: Vec<RerankInput> = results
|
||||
.iter()
|
||||
.map(|r| RerankInput {
|
||||
index: r.index,
|
||||
timestamp: r.timestamp,
|
||||
source_channel: r.source_channel.clone(),
|
||||
raw_activation: r.activation,
|
||||
relevance: r.score,
|
||||
})
|
||||
.collect();
|
||||
let order: Vec<usize> = rerank(&inputs, &config, now)
|
||||
.into_iter()
|
||||
.map(|r| r.index)
|
||||
.collect();
|
||||
let by_index: HashMap<usize, SearchResult> =
|
||||
results.into_iter().map(|r| (r.index, r)).collect();
|
||||
results = order
|
||||
.into_iter()
|
||||
.filter_map(|i| by_index.get(&i).cloned())
|
||||
.collect();
|
||||
}
|
||||
results.truncate(top_k);
|
||||
let latency = t0.elapsed();
|
||||
|
||||
// Rank of the best-placed result from each gold session.
|
||||
let mut first_rank: HashMap<&str, usize> = HashMap::new();
|
||||
for (rank, result) in results.iter().enumerate() {
|
||||
let sid = memory.cache.session_ids[result.index].as_str();
|
||||
if let Some((gold_sid, _)) = gold_times.get_key_value(sid) {
|
||||
first_rank.entry(gold_sid).or_insert(rank);
|
||||
}
|
||||
}
|
||||
let newest_gold_first = if gold_times.len() < 2 || first_rank.is_empty() {
|
||||
None
|
||||
} else {
|
||||
// The newest gold session must be retrieved, and no older gold session
|
||||
// may outrank it.
|
||||
let newest = gold_times
|
||||
.iter()
|
||||
.max_by(|a, b| a.1.total_cmp(b.1))
|
||||
.map(|(sid, _)| *sid)
|
||||
.expect("at least two gold sessions");
|
||||
Some(match first_rank.get(newest) {
|
||||
Some(&newest_rank) => first_rank
|
||||
.iter()
|
||||
.all(|(sid, &rank)| *sid == newest || rank > newest_rank),
|
||||
None => false,
|
||||
})
|
||||
};
|
||||
|
||||
// Session-level recall
|
||||
let mut hit1_session = false;
|
||||
let mut hit5_session = false;
|
||||
@@ -415,6 +617,7 @@ fn evaluate_question(
|
||||
hit5_turn,
|
||||
hit10_turn,
|
||||
rr_turn,
|
||||
newest_gold_first,
|
||||
latency,
|
||||
}
|
||||
}
|
||||
@@ -566,6 +769,24 @@ fn print_report(
|
||||
);
|
||||
println!();
|
||||
|
||||
if let Some(pct) = overall.newest_gold_first_pct() {
|
||||
println!(
|
||||
"## Recency Discrimination (n={})",
|
||||
overall.newest_gold_total
|
||||
);
|
||||
println!(
|
||||
" Newest gold session ranked first: {}/{} ({pct:.1}%)",
|
||||
overall.newest_gold_first, overall.newest_gold_total
|
||||
);
|
||||
println!(
|
||||
" Questions whose evidence spans several dated sessions — a fact and\n \
|
||||
its later correction. Both sessions are labelled gold, so recall\n \
|
||||
scores either as a hit; this asks whether the *current* one came\n \
|
||||
first. A retriever with no sense of time scores near chance."
|
||||
);
|
||||
println!();
|
||||
}
|
||||
|
||||
if overall.abstention_total > 0 {
|
||||
println!("## Abstention Accuracy");
|
||||
println!(
|
||||
@@ -679,6 +900,14 @@ fn print_report(
|
||||
} else {
|
||||
println!(" \"abstention_accuracy\": null,");
|
||||
}
|
||||
match overall.newest_gold_first_pct() {
|
||||
Some(pct) => println!(
|
||||
" \"newest_gold_first\": {:.4}, \"newest_gold_n\": {},",
|
||||
pct / 100.0,
|
||||
overall.newest_gold_total
|
||||
),
|
||||
None => println!(" \"newest_gold_first\": null,"),
|
||||
}
|
||||
println!(" \"latency_us\": {{");
|
||||
println!(
|
||||
" \"avg\": {:.1}, \"p50\": {:.1}, \"p95\": {:.1}, \"p99\": {:.1}",
|
||||
@@ -701,6 +930,8 @@ fn main() {
|
||||
let mut limit: Option<usize> = None;
|
||||
let mut weights_dir: Option<String> = None;
|
||||
let mut sweep = false;
|
||||
#[cfg_attr(not(feature = "embeddings"), allow(unused_mut, unused_variables))]
|
||||
let mut rerank_sweep = false;
|
||||
let mut args = std::env::args().skip(1);
|
||||
while let Some(arg) = args.next() {
|
||||
match arg.as_str() {
|
||||
@@ -709,6 +940,16 @@ fn main() {
|
||||
limit = Some(v.parse().expect("--limit must be a positive integer"));
|
||||
}
|
||||
"--sweep" => sweep = true,
|
||||
"--rerank-sweep" => {
|
||||
// Re-ranking needs the vector stage to have candidates worth
|
||||
// reordering, so this is an embeddings-only comparison.
|
||||
#[cfg(feature = "embeddings")]
|
||||
{
|
||||
rerank_sweep = true;
|
||||
}
|
||||
#[cfg(not(feature = "embeddings"))]
|
||||
eprintln!("warning: --rerank-sweep needs --features embeddings; ignoring");
|
||||
}
|
||||
"--embeddings" => {
|
||||
weights_dir = Some(args.next().expect("--embeddings needs a directory"));
|
||||
}
|
||||
@@ -727,6 +968,9 @@ fn main() {
|
||||
BM25-only, vector-only, and hybrid separately. Requires\n\
|
||||
--features embeddings; without it the vector stage is\n\
|
||||
inert and only the BM25 row is produced.\n\
|
||||
--rerank-sweep\n\
|
||||
compare re-ranking off, metadata-only (the old\n\
|
||||
behaviour) and blended at several half-lives.\n\
|
||||
--sweep instead of the three named modes, sweep vector_weight\n\
|
||||
from 0.0 to 1.0 in 0.1 steps. The 0.7/0.3 default was\n\
|
||||
never searched; this is what searches it."
|
||||
@@ -792,6 +1036,10 @@ fn main() {
|
||||
{
|
||||
if sweep {
|
||||
sweep_modes()
|
||||
} else if rerank_sweep {
|
||||
let mut modes = vec![HYBRID, hybrid_rerank_metadata_only()];
|
||||
modes.extend(hybrid_rerank_half_lives());
|
||||
modes
|
||||
} else {
|
||||
vec![
|
||||
BM25_ONLY,
|
||||
@@ -800,6 +1048,8 @@ fn main() {
|
||||
RRF,
|
||||
BM25_STEMMED,
|
||||
HYBRID_STEMMED,
|
||||
hybrid_rerank_metadata_only(),
|
||||
hybrid_rerank_blended(),
|
||||
]
|
||||
}
|
||||
}
|
||||
@@ -916,6 +1166,14 @@ fn run_mode(
|
||||
entry.rr_turn += rr;
|
||||
overall.rr_turn += rr;
|
||||
}
|
||||
if let Some(newest_first) = result.newest_gold_first {
|
||||
entry.newest_gold_total += 1;
|
||||
overall.newest_gold_total += 1;
|
||||
if newest_first {
|
||||
entry.newest_gold_first += 1;
|
||||
overall.newest_gold_first += 1;
|
||||
}
|
||||
}
|
||||
|
||||
let ns = result.latency.as_nanos() as u64;
|
||||
entry.latency_ns.push(ns);
|
||||
@@ -927,3 +1185,30 @@ fn run_mode(
|
||||
eprintln!();
|
||||
print_report(&overall, &by_type, profile, mode);
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::session_time;
|
||||
|
||||
#[test]
|
||||
fn session_dates_parse_to_the_right_instant() {
|
||||
// Reference values from Python's datetime, UTC.
|
||||
for (date, expected) in [
|
||||
("2023/05/25 (Thu) 20:21", 1_685_046_060.0),
|
||||
("1970/01/01 (Thu) 00:00", 0.0),
|
||||
("2000/02/29 (Tue) 12:00", 951_825_600.0),
|
||||
("2023/12/31 (Sun) 23:59", 1_704_067_140.0),
|
||||
("2024/03/01 (Fri) 00:00", 1_709_251_200.0),
|
||||
] {
|
||||
assert_eq!(session_time(date, 0), expected, "{date}");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unparseable_dates_fall_back_to_position_order() {
|
||||
let a = session_time("not a date", 0);
|
||||
let b = session_time("", 1);
|
||||
let c = session_time("2023/13/99 (???) 99:99", 2);
|
||||
assert!(a < b && b < c, "fallback must preserve session order");
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user