prototype implementation for extensible imbalance scheme
(cherry picked from commit 362a26a3def0803546b986f951bf9868acc0db15)