performance tuning - How to flush machine underflows to zero and prevent conversion to arbitrary precision?
I'm working on some pretty intense computation in Mathematica; when my code started running slowly, I tracked the source of the problem to Exp[]
. I need to exponentiate every element of a 50x500x500 array; performing the operation on a 500x500 array takes on the order of 3 seconds (according to AbsoluteTime
), so the entire array should take about 50 times that. Unfortunately, that's calculation needs to happen for every data point.
I've read about lots of ways to speed up Mathematica code, but none of those methods seem to apply here. I'm already working in MachinePrecision
. I have noticed that some of my results are ridiculously small (for example, 4.282835067271648*10^-78127094
), but I'm not sure how to make Mathematica ignore those; they're obviously much smaller than $MachineEpsilon
.
Any advice is greatly appreciated!
Update:
Below is a sample of my code and the generated output. To give it some context, g0
, is a scalar, σg0
is a length 50 array, and g
is a 500x500 array.
(* Added after Oleksandr R.'s comment *)
SetSystemOptions["CatchMachineUnderflow" -> False];
n = Length[σg0];
probgs = ConstantArray[N[0], {50, 500, 500}];
For[i = 1, i <= n, i++,
probgs[[i]] =
N[(1/(Sqrt[2 π] σg0[[i]])) Exp[-0.5 ((g - g0)/σg0[[i]])^2]];
]; // AbsoluteTiming
Precision[probgs]
Output:
{4.816275, Null}
MachinePrecision
Turning off underflow definitely helped; 5 seconds isn't bad at all for what I'm doing.
Answer
Obviously, for large negative inputs, Exp
will produce very small numbers. While this isn't intrinsically problematic, it so happens that, by default, Mathematica deals with machine underflow by converting the affected values to an arbitrary precision representation in order to avoid catastrophic loss of precision. However, sometimes one would rather disregard underflowed values instead (i.e. let them go to zero), and indeed that seems to be the case here.
This behavior can be controlled using the system option "CatchMachineUnderflow"
--simply use
SetSystemOptions["CatchMachineUnderflow" -> False]
and underflowed values will be flushed to (machine precision) zero.
Since this is a global option that will most likely affect the results of system functions as well as user code, it's advisable to localize its effect as tightly as possible. For this purpose one can use the undocumented function Internal`WithLocalSettings
, as described by Daniel Lichtblau in this StackOverflow answer:
With[{cmuopt = SystemOptions["CatchMachineUnderflow"]},
Internal`WithLocalSettings[
SetSystemOptions["CatchMachineUnderflow" -> False],
(* put your own code here; for example: *)
Exp[-1000.],
SetSystemOptions[cmuopt]
]
]
(* 0.` *)
Contrast this with:
Exp[-1000.]
(* 5.0759588975494567652918094795743369258164499728`12.954589770191006*^-435 *)
Comments
Post a Comment