Storage capacity optimization of non-volatile memory is shown. Through a controller, communication between a host and a non-volatile memory is in units of a first data length. The controller manages a bad column table for the non-volatile memory in units of a second data length. The second data length is shorter than the first data length. Taking byte communication as an example, one nibble of storage units is marked as bad when it has any damaged storage units.