add probabilistic weight module