From 46a8601140d3e1891316c93f8ea4ed4c73fe7f55 Mon Sep 17 00:00:00 2001
From: Francis Couture-Harpin <git@compilade.net>
Date: Tue, 5 Aug 2025 13:34:01 -0400
Subject: [PATCH] quantize : assume the neutral prior is equal imatrix weights

---
 tools/quantize/quantize.cpp | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tools/quantize/quantize.cpp b/tools/quantize/quantize.cpp
index a6a528b74a..6846850bdd 100644
--- a/tools/quantize/quantize.cpp
+++ b/tools/quantize/quantize.cpp
@@ -291,8 +291,15 @@ static int load_imatrix(const std::string & imatrix_file, std::vector<std::strin
         for (int64_t j = 0; j < ne1; ++j) {
             const float count = ((const float *) counts->data)[j];
             if (count > 0.0f) {
+                float sumw = 0.0f;
                 for (int64_t i = 0; i < ne0; ++i) {
-                    e[j*ne0 + i] = (((const float *) sums->data)[j*ne0 + i] + prior_weight) / (count + prior_weight);
+                    sumw += ((const float *) sums->data)[j*ne0 + i];
+                }
+                // the neutral prior is equal weights, and it should reduce the variance by weighted-averaging with the mean
+                const float prior_value = sumw / ne0;
+
+                for (int64_t i = 0; i < ne0; ++i) {
+                    e[j*ne0 + i] = (((const float *) sums->data)[j*ne0 + i] + prior_value * prior_weight) / (count + prior_weight);
                 }
             } else {
                 // Partial imatrix data, this tensor never got any input during calibration