Note, however, that the LP relaxation optimized by dual decomposition is significantly tighter than Z.
an alternative approach would be to use the dual decomposition algorithm for inference during training.