首页  >  文章  >  后端开发  >  FP和BF有什么区别这里给你很好的解释

FP和BF有什么区别这里给你很好的解释

WBOY
WBOY原创
2024-07-30 00:03:43873浏览

What is the difference between FPand BF Here a good explanation for you

浮点表示:

FP16(半精度):在 FP16 中,浮点数使用 16 位表示。它由 1 个符号位、5 位指数和 10 位分数(尾数)组成。这种格式可以提供更高的精度来表示其范围内的小数值。

BF16 (BFloat16):BF16 也使用 16 位,但分布不同。它有 1 个符号位、8 位指数、7 位尾数。这种格式牺牲了小数部分的一些精度以适应更广泛的指数。

数值范围:

FP16 由于其 10 位尾数,其范围较小,但在该范围内精度较高。
BF16 由于其 8 位指数和 7 位尾数,其小数范围较宽,但精度较低。

示例:

我们用3个例子来说明FP16和BF16的区别。使用TensorFlow来做底层的测试和代码共享:

原始值:0.0001 — 两种方法都可以表示
FP16: 0.00010001659393(二进制:0|00001|1010001110,十六进制:068E) — 10 个尾数和 5 个指数
BF16:0.00010013580322(二进制:0|01110001|1010010,十六进制:38D2)— 7 个尾数和 8 个指数

如您所见,它们具有不同的指数和尾数,因此能够表示不同的值。 但我们可以看到 FP16 表示得更准确,值更接近。

原始值:1e-08 (0.00000001)
FP16:0.0000000000000(二进制:0|00000|0000000000,十六进制:0000)
BF16:0.000000010 01172(二进制:0|01100100| 0101100,十六进制:322C)

这是一个非常有趣的案例。 FP16 失败 并使结果为 0,但 BF16 能够用特殊格式表示它。

原始值:100000.00001
FP16:inf(二进制:0|11111|0000000000,十六进制:7C00)
BF16:99840.00000000000000(二进制:0|100) 01111|1000011,十六进制:47C3 )

在上述情况下,FP16 失败,因为所有指数位都已满并且不足以表示该值。然而 BF16 有效

用例:

FP16 常用于深度学习训练和推理,特别是对于需要高精度表示有限范围内的小分数值的任务。

BF16 在为机器学习任务设计的硬件架构中变得越来越流行,这些任务受益于更广泛的可表示值,即使是以小数部分的一些精度为代价。当处理大梯度或大范围内的数值稳定性比小值的精度更重要时,它特别有用。

总结

FP16 为较小范围内的小数值提供了更高的精度,使其适合需要精确表示小数字的任务。另一方面,BF16 以牺牲一定精度为代价提供了更广泛的范围,这使得它有利于涉及更广泛值范围或在大范围内的数值稳定性至关重要的任务。 FP16 和 BF16 之间的选择取决于手头机器学习任务的具体要求。

最终结论

由于以上原因,在进行 Stable Diffusion XL (SDXL) 训练时,FP16 和 BF16 需要的学习率略有不同,我发现 BF16 效果更好。

用于生成上述示例的代码

import tensorflow as tf
import struct

def float_to_binary(f):
    return ''.join(f'{b:08b}' for b in struct.pack('>f', f))

def display_fp16(value):
    fp16 = tf.cast(tf.constant(value, dtype=tf.float32), tf.float16)
    fp32 = tf.cast(fp16, tf.float32)
    binary = format(int.from_bytes(fp16.numpy().tobytes(), 'big'), '016b')
    sign = binary[0]
    exponent = binary[1:6]
    fraction = binary[6:]
    return f"FP16: {fp32.numpy():14.14f} (Binary: {sign}|{exponent}|{fraction}, Hex: {fp16.numpy().view('uint16'):04X})"

def display_bf16(value):
    bf16 = tf.cast(tf.constant(value, dtype=tf.float32), tf.bfloat16)
    bf32 = tf.cast(bf16, tf.float32)
    binary = format(int.from_bytes(bf16.numpy().tobytes(), 'big'), '016b')
    sign = binary[0]
    exponent = binary[1:9]
    fraction = binary[9:]
    return f"BF16: {bf32.numpy():14.14f} (Binary: {sign}|{exponent}|{fraction}, Hex: {bf16.numpy().view('uint16'):04X})"

values = [0.0001, 0.00000001, 100000.00001]

for value in values:
    print(f"\nOriginal value: {value}")
    print(display_fp16(value))
    print(display_bf16(value))

以上是FP和BF有什么区别这里给你很好的解释的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn