smp-tee-runtime 0.1.0

Hardened minimal runtime for TEE-based federated aggregation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

pub fn federated_averaging(vectors: &[Vec<f32>]) -> Option<Vec<f32>> {
    let dimension = vectors.first()?.len();
    if vectors.iter().any(|vector| vector.len() != dimension) {
        return None;
    }

    let mut acc = vec![0.0_f32; dimension];
    for vector in vectors {
        for (sum, value) in acc.iter_mut().zip(vector.iter()) {
            *sum += value;
        }
    }

    let denom = vectors.len() as f32;
    acc.iter_mut().for_each(|sum| *sum /= denom);
    Some(acc)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn federated_averaging_returns_mean() {
        let avg = federated_averaging(&[vec![1.0, 2.0], vec![3.0, 4.0], vec![5.0, 6.0]]).unwrap();
        assert_eq!(avg, vec![3.0, 4.0]);
    }
}