improving training with probability and sampling theory