We use a tetra soft bed mattress in DS cot as he has issues with the hard mattress too. It makes him cosy as he is not so exposed in the cot but still able to get used to being in the cot ITMS.

Sorry I have no other solutions to offer, DS falls asleep in my arms or after a feed most of the time and does not notice the transition from arms to bed this way.