US giant Albemarle will switch off its new lithium asset in Western Australia's Pilbara region if demand for the battery commodity proves to be weak.

Albemarle increased its exposure to Australia in December when it agreed to pay $US1.15 billion ($1.64 billion) for half of Mineral Resources Ltd's Wodgina lithium mine near Port Hedland.