Part 2 request for review comments.

Also, high-level refactoring (explicit places) was carried out for the entire module.

More detailed refactoring will continue with the "validator_logits_divergence" method.