This loss function will be used in derivations later in this work.
Note that we have dropped the long entropy term for brevity as it is not relevant to our derivations, but it can be viewed in the formulation by [1]}.
