We approximate the marginalization by moving the outside expectation directly inside over A.
we instead optimize the following objective: where E[A] are the marginals of A with respect to p (A|x, t, h).