I don't think the shrink problem is at all the same for the two technologies. There are some really weird materials and production steps in Optane that are simply not present when making Flash cells.

durability drops quickly with shrinking flash, we won't see much smaller cells, the growth has been MLC-TLC-> QLC and stacking