Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 2 additions & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -233,6 +233,8 @@ members = [
"crates/ruvllm_retrieval_diffusion",
# RAIRS IVF: Redundant Assignment + Amplified Inverse Residual (ADR-193)
"crates/ruvector-rairs",
# MinCut-guided agent memory compaction (ADR-196)
"crates/ruvector-mincut-memory",
]
resolver = "2"

Expand Down
30 changes: 30 additions & 0 deletions crates/ruvector-mincut-memory/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
[package]
name = "ruvector-mincut-memory"
version = "0.1.0"
edition = "2021"
description = "MinCut-guided agent working memory compaction for RuVector — graph-cut coherence eviction"
authors = ["ruvnet", "claude-flow"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/ruvnet/ruvector"
keywords = ["agent-memory", "graph-cut", "vector-search", "compaction", "ruvector"]
categories = ["algorithms", "data-structures"]

[[bin]]
name = "benchmark"
path = "src/main.rs"

[dependencies]
rand = "0.8"
rand_distr = "0.4"
serde = { version = "1", features = ["derive"] }

[dev-dependencies]
criterion = { version = "0.5", features = ["html_reports"] }

[[bench]]
name = "compaction_bench"
harness = false

[lints.rust]
dead_code = "allow"
unused_variables = "allow"
62 changes: 62 additions & 0 deletions crates/ruvector-mincut-memory/benches/compaction_bench.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
use rand::{Rng, SeedableRng};
use rand::rngs::StdRng;
use rand_distr::{Distribution, Normal};
use ruvector_mincut_memory::{AgeEvict, CoherenceEvict, Compactor, MemoryStore, MinCutEvict};

fn gen_store(n: usize, dims: usize, clusters: usize, threshold: f32, seed: u64) -> MemoryStore {
let mut rng = StdRng::seed_from_u64(seed);
let normal = Normal::new(0.0f32, 0.25).unwrap();
let centroids: Vec<Vec<f32>> = (0..clusters)
.map(|_| {
let raw: Vec<f32> = (0..dims).map(|_| rng.gen::<f32>() * 2.0 - 1.0).collect();
let norm: f32 = raw.iter().map(|x| x * x).sum::<f32>().sqrt().max(1e-9);
raw.iter().map(|x| x / norm).collect()
})
.collect();

let mut store = MemoryStore::new(dims, threshold);
for ts in 0..n {
let c = rng.gen_range(0..clusters);
let mut v: Vec<f32> = centroids[c]
.iter()
.map(|x| x + normal.sample(&mut rng))
.collect();
let norm: f32 = v.iter().map(|x| x * x).sum::<f32>().sqrt().max(1e-9);
v.iter_mut().for_each(|x| *x /= norm);
store.insert(v, ts as u64);
}
store
}

fn bench_compaction(c: &mut Criterion) {
let dims = 32usize;
let threshold = 0.4f32;
let clusters = 6usize;

let strategies: Vec<(&str, Box<dyn Compactor>)> = vec![
("AgeEvict", Box::new(AgeEvict)),
("CoherenceEvict", Box::new(CoherenceEvict)),
("MinCutEvict", Box::new(MinCutEvict)),
];

let mut group = c.benchmark_group("compaction");

for n in [100usize, 300usize] {
let target = n / 2;
for (name, strat) in &strategies {
group.bench_with_input(BenchmarkId::new(*name, n), &n, |b, &n| {
b.iter_batched(
|| gen_store(n, dims, clusters, threshold, 42),
|mut store| strat.compact(&mut store, target),
criterion::BatchSize::SmallInput,
);
});
}
}

group.finish();
}

criterion_group!(benches, bench_compaction);
criterion_main!(benches);
Loading
Loading