May want to try in:
Settings > Display > Screen Timeout
Adjust these settings and see if you have any change. I don't have a Samsung but I do have the problem of my wifi dropping. On my android, my wifi stays longer if I set the timeout to a higher value.