Memory modeling of counterfactual generation