CSI feedback overhead reduction using Learned Approximate Message Passing(LAMP)

I have the input that has to compressed using LAMP, a neural network Algorithm. The output should be compressed and recovered again. This is wireless communication project on 5G where the feedback overhead has to be reduce while sending the CSI from UE(User Equipment) to gNb(Base station). Attached is the paper for reference.