Struct CodebookBuilder 

pub struct CodebookBuilder { /* private fields */ }

Multi-stage Vector Codebook Builder implementing Stage 2 of NOVAQ

Mathematical formulation: For each subspace k:

1. Coarse quantization: b^(1){i,k} = argmin_c ||v{i,k} - C^(1)_{c,k}||²
2. Residual calculation: r_{i,k} = v_{i,k} - C^(1){b^(1){i,k},k}
3. Residual quantization: b^(2){i,k} = argmin_c ||r{i,k} - C^(2)_{c,k}||²

Final reconstruction: W_{i,:} = Σ_{k=1}^N (C^(1){b^(1){i,k},k} + C^(2){b^(2){i,k},k})

Implementations

impl CodebookBuilder

pub fn new( num_subspaces: usize, codebook_size_l1: usize, codebook_size_l2: usize, seed: u64, ) -> Self

pub fn build_codebooks( &mut self, weights: &WeightMatrix, ) -> Result<(VectorCodebooks, QuantizationIndices)>

Build multi-stage vector codebooks for given weight matrix with enhanced stability

pub fn reconstruct_weights( &self, codebooks: &VectorCodebooks, indices: &QuantizationIndices, rows: usize, cols: usize, ) -> Result<Vec<f32>>

Reconstruct weights from codebooks and indices

Trait Implementations

impl Debug for CodebookBuilder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.